-
Notifications
You must be signed in to change notification settings - Fork 75
/
cfg.py
68 lines (66 loc) · 3.11 KB
/
cfg.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
config = {
'bsds500': {'data_root': 'path_to/bsds500/HED-BSDS/',
'data_lst': 'train_pair.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.5},
'pascal_context': {'data_root': 'path_to/bsds500/PASCAL/',
'data_lst': 'voc_trainval.txt',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.3},
'bsds_pascal': {'data_root': 'path_to/bsds500/',
'data_lst': 'bsds_pascal_train_pair.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.5},
'NYUDv2_HHA': {'data_root': 'path_to/NYUD/',
'data_lst': 'hha-train.lst',
'mean_bgr': [109.92, 88.24, 127.42],
'yita': 0.5},
'NYUDv2_RGB': {'data_root': 'path_to/NYUD/',
'data_lst': 'image-train.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.5},
'MulticueEdges': {'data_root': 'path_to/multicue/',
'data_lst': 'train_edges_aug%d.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.3},
'MulticueBoundaries': {'data_root': 'path_to/multicue/',
'data_lst': 'train_boundaries_aug%d.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.4}
}
config_test = {
'bsds500': {'data_root': 'path_to/bsds500/BSR/BSDS500/data/',
'data_lst': 'test_pair.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.5},
'bsds500_val': {'data_root': 'path_to/bsds500/BSR/BSDS500/data/',
'data_lst': 'val_pair.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.5},
'pascal_context': {'data_root': 'path_to/bsds500/PASCAL/',
'data_lst': 'voc_test.txt',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.3},
'pascal_context_journal_val': {'data_root': 'path_to/bsds500/PASCAL/',
'data_lst': 'voc_validation_pair.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 1},
'NYUDv2_HHA': {'data_root': 'path_to/NYUD/',
'data_lst': 'hha-test.lst',
'mean_bgr': [109.92, 88.24, 127.42],
'yita': 0.5},
'NYUDv2_RGB': {'data_root': 'path_to/NYUD/',
'data_lst': 'image-test.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.5},
'MulticueEdges': {'data_root': 'path_to/multicue/',
'data_lst': 'test%d.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.3},
'MulticueBoundaries': {'data_root': 'path_to/multicue/',
'data_lst': 'test%d.lst',
'mean_bgr': [104.00699, 116.66877, 122.67892],
'yita': 0.4}
}
if __name__ == '__main__':
print config.keys()