forked from Ellinier/DSOD-Pytorch-Implementation
-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
84 lines (72 loc) · 2.72 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
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
from easydict import EasyDict as edict
cfg = edict()
cfg.CLASSES_voc = (
'aeroplane',
'bicycle',
'bird',
'boat',
'bottle',
'bus',
'car',
'cat',
'chair',
'cow',
'diningtable',
'dog',
'horse',
'motorbike',
'person',
'pottedplant',
'sheep',
'sofa',
'train',
'tvmonitor',
)
cfg.CLASSES_ours_15cls = (
'pencil',
'highlighter',
'penholder',
'tape',
'soap',
'scissors',
'dumbbells',
'toothbrush',
'bottle',
'indexcards',
'boarderaser',
'glue',
'baseball',
'screwdriver',
'crayon',
)
cfg.VOC07images = '/opt/VOCdevkit/VOC2007/JPEGImages/'
cfg.list_file_train_voc07 = './label_list/voc07_train.txt'
cfg.list_file_test_voc07 = './label_list/voc07_test.txt'
cfg.VOC12images = '/opt/VOCdevkit/VOC2012/JPEGImages/'
cfg.list_file_train_voc12 = './label_list/voc12_train.txt'
cfg.list_file_test_voc12 = './label_list/voc12_test.txt'
cfg.ours_15cls_images = '/home/share/dataset/VOC2007For15Cls/JPEGImages/'
cfg.list_file_train_ours_15cls = './label_list/ours_15cls_train.txt'
cfg.list_file_test_ours_15cls = './label_list/ours_15cls_test.txt'
cfg.ours_15cls_4300_images = '/home/share/dataset/VOC2007For4300/JPEGImages/'
cfg.list_file_train_ours_15cls_4300 = './label_list/ours_15cls_4300_train.txt'
cfg.list_file_train_ours_15cls_4300 = './label_list/ours_15cls_4300_train3.txt'
cfg.list_file_test_ours_15cls_4300 = './label_list/ours_15cls_4300_test.txt'
cfg.ours_15cls_4600_images = '/home/share/dataset/VOC2007For4600/JPEGImages/'
cfg.list_file_train_ours_15cls_4600 = './label_list/ours_15cls_4600_train.txt'
#cfg.list_file_train_ours_15cls_4600 = './label_list/ours_15cls_4600_train2.txt'
cfg.list_file_test_ours_15cls_4600 = './label_list/ours_15cls_4600_test.txt'
cfg.ours_15cls_4300_Tmp_50_images = '/home/share/dataset/VOC2007For4300_Tmp_50/JPEGImages/'
cfg.list_file_train_ours_15cls_4300_Tmp_50 = './label_list/ours_15cls_4300_Tmp_50_train3.txt'
#cfg.list_file_train_ours_15cls_4600 = './label_list/ours_15cls_4600_train2.txt'
cfg.list_file_test_ours_15cls_4300_Tmp_50 = './label_list/ours_15cls_4300_test.txt'
cfg.supplement100_images = '/home/share/dataset/supplement100/JPEGImages/'
cfg.list_file_train_supplement100 = './label_list/supplement100_train.txt'
cfg.list_file_test_supplement100 = './label_list/supplement100_test.txt'
cfg.supplement100_images = '/home/ellin/Downloads/supplement100/JPEGImages/'
cfg.list_file_train_supplement100 = './label_list/supplement100_train.txt'
cfg.list_file_test_supplement100 = './label_list/supplement100_test.txt'
cfg.CLASSES = cfg.CLASSES_ours_15cls
cfg.img_root = cfg.supplement100_images
cfg.label_train = cfg.list_file_train_supplement100
cfg.label_test = cfg.list_file_test_supplement100