-
Notifications
You must be signed in to change notification settings - Fork 100
/
Copy pathdataset_config.py
101 lines (98 loc) · 4.78 KB
/
dataset_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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
from os.path import join
_BASE_DATA_PATH = "../data"
dataset_config = {
'mnist': {
'path': join(_BASE_DATA_PATH, 'mnist'),
'normalize': ((0.1307,), (0.3081,)),
# Use the next 3 lines to use MNIST with a 3x32x32 input
# 'extend_channel': 3,
# 'pad': 2,
# 'normalize': ((0.1,), (0.2752,)) # values including padding
},
'svhn': {
'path': join(_BASE_DATA_PATH, 'svhn'),
'resize': (224, 224),
'crop': None,
'flip': False,
'normalize': ((0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
},
'cifar100': {
'path': join(_BASE_DATA_PATH, 'cifar100'),
'resize': None,
'pad': 4,
'crop': 32,
'flip': True,
'normalize': ((0.5071, 0.4866, 0.4409), (0.2009, 0.1984, 0.2023))
},
'cifar100_icarl': {
'path': join(_BASE_DATA_PATH, 'cifar100'),
'resize': None,
'pad': 4,
'crop': 32,
'flip': True,
'normalize': ((0.5071, 0.4866, 0.4409), (0.2009, 0.1984, 0.2023)),
'class_order': [
68, 56, 78, 8, 23, 84, 90, 65, 74, 76, 40, 89, 3, 92, 55, 9, 26, 80, 43, 38, 58, 70, 77, 1, 85, 19, 17, 50,
28, 53, 13, 81, 45, 82, 6, 59, 83, 16, 15, 44, 91, 41, 72, 60, 79, 52, 20, 10, 31, 54, 37, 95, 14, 71, 96,
98, 97, 2, 64, 66, 42, 22, 35, 86, 24, 34, 87, 21, 99, 0, 88, 27, 18, 94, 11, 12, 47, 25, 30, 46, 62, 69,
36, 61, 7, 63, 75, 5, 32, 4, 51, 48, 73, 93, 39, 67, 29, 49, 57, 33
]
},
'vggface2': {
'path': join(_BASE_DATA_PATH, 'VGGFace2'),
'resize': 256,
'crop': 224,
'flip': True,
'normalize': ((0.5199, 0.4116, 0.3610), (0.2604, 0.2297, 0.2169))
},
'imagenet_256': {
'path': join(_BASE_DATA_PATH, 'ILSVRC12_256'),
'resize': None,
'crop': 224,
'flip': True,
'normalize': ((0.485, 0.456, 0.406), (0.229, 0.224, 0.225))
},
'imagenet_subset': {
'path': join(_BASE_DATA_PATH, 'ILSVRC12_256'),
'resize': None,
'crop': 224,
'flip': True,
'normalize': ((0.485, 0.456, 0.406), (0.229, 0.224, 0.225)),
'class_order': [
68, 56, 78, 8, 23, 84, 90, 65, 74, 76, 40, 89, 3, 92, 55, 9, 26, 80, 43, 38, 58, 70, 77, 1, 85, 19, 17, 50,
28, 53, 13, 81, 45, 82, 6, 59, 83, 16, 15, 44, 91, 41, 72, 60, 79, 52, 20, 10, 31, 54, 37, 95, 14, 71, 96,
98, 97, 2, 64, 66, 42, 22, 35, 86, 24, 34, 87, 21, 99, 0, 88, 27, 18, 94, 11, 12, 47, 25, 30, 46, 62, 69,
36, 61, 7, 63, 75, 5, 32, 4, 51, 48, 73, 93, 39, 67, 29, 49, 57, 33
]
},
'imagenet_32_reduced': {
'path': join(_BASE_DATA_PATH, 'ILSVRC12_32'),
'resize': None,
'pad': 4,
'crop': 32,
'flip': True,
'normalize': ((0.481, 0.457, 0.408), (0.260, 0.253, 0.268)),
'class_order': [
472, 46, 536, 806, 547, 976, 662, 12, 955, 651, 492, 80, 999, 996, 788, 471, 911, 907, 680, 126, 42, 882,
327, 719, 716, 224, 918, 647, 808, 261, 140, 908, 833, 925, 57, 388, 407, 215, 45, 479, 525, 641, 915, 923,
108, 461, 186, 843, 115, 250, 829, 625, 769, 323, 974, 291, 438, 50, 825, 441, 446, 200, 162, 373, 872, 112,
212, 501, 91, 672, 791, 370, 942, 172, 315, 959, 636, 635, 66, 86, 197, 182, 59, 736, 175, 445, 947, 268,
238, 298, 926, 851, 494, 760, 61, 293, 696, 659, 69, 819, 912, 486, 706, 343, 390, 484, 282, 729, 575, 731,
530, 32, 534, 838, 466, 734, 425, 400, 290, 660, 254, 266, 551, 775, 721, 134, 886, 338, 465, 236, 522, 655,
209, 861, 88, 491, 985, 304, 981, 560, 405, 902, 521, 909, 763, 455, 341, 905, 280, 776, 113, 434, 274, 581,
158, 738, 671, 702, 147, 718, 148, 35, 13, 585, 591, 371, 745, 281, 956, 935, 346, 352, 284, 604, 447, 415,
98, 921, 118, 978, 880, 509, 381, 71, 552, 169, 600, 334, 171, 835, 798, 77, 249, 318, 419, 990, 335, 374,
949, 316, 755, 878, 946, 142, 299, 863, 558, 306, 183, 417, 64, 765, 565, 432, 440, 939, 297, 805, 364, 735,
251, 270, 493, 94, 773, 610, 278, 16, 363, 92, 15, 593, 96, 468, 252, 699, 377, 95, 799, 868, 820, 328, 756,
81, 991, 464, 774, 584, 809, 844, 940, 720, 498, 310, 384, 619, 56, 406, 639, 285, 67, 634, 792, 232, 54,
664, 818, 513, 349, 330, 207, 361, 345, 279, 549, 944, 817, 353, 228, 312, 796, 193, 179, 520, 451, 871,
692, 60, 481, 480, 929, 499, 673, 331, 506, 70, 645, 759, 744, 459]
}
}
# Add missing keys:
for dset in dataset_config.keys():
for k in ['resize', 'pad', 'crop', 'normalize', 'class_order', 'extend_channel']:
if k not in dataset_config[dset].keys():
dataset_config[dset][k] = None
if 'flip' not in dataset_config[dset].keys():
dataset_config[dset]['flip'] = False