forked from barongeng/mask_rcnn_pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
32 lines (31 loc) · 1.17 KB
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
# this is used for storing configurations of datasets & models
datasets = {
'coco-trainval35k-minival': {
'num_classes': 80,
'scale_size': 800,
'train_split': 'annotations/instances_trainval35k2014.json',
'val_split': 'annotations/instances_minival2014.json',
'test_split': 'annotations/image_info_test-dev2015.json',
},
'coco-train-minival': {
'num_classes': 80,
'scale_size': 800,
'train_split': 'annotations/instances_train2014.json',
'val_split': 'annotations/instances_minival2014.json',
'test_split': 'annotations/image_info_test-dev2015.json',
},
'coco-train-val': {
'num_classes': 80,
'scale_size': 800,
'train_split': 'annotations/instances_train2014.json',
'val_split': 'annotations/instances_val2014.json',
'test_split': 'annotations/image_info_test-dev2015.json',
},
'coco-debug': {
'num_classes': 80,
'scale_size': 600,
'train_split': 'annotations/instances_minival2014.json',
'val_split': 'annotations/instances_minival2014.json',
'test_split': 'annotations/image_info_test-dev2015.json',
},
}