forked from eladhoffer/ImageNet-Training
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathConfig.lua
More file actions
23 lines (20 loc) · 884 Bytes
/
Config.lua
File metadata and controls
23 lines (20 loc) · 884 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
local ImageNetClasses = torch.load('./ImageNetClasses')
for i=1001,#ImageNetClasses.ClassName do
ImageNetClasses.ClassName[i] = nil
end
function Key(num)
return string.format('%07d',num)
end
return
{
TRAINING_PATH = '/home/ehoffer/Datasets/ImageNet/train/', --Training images location
VALIDATION_PATH = '/home/ehoffer/Datasets/ImageNet/validation/', --Validation images location
VALIDATION_DIR = '/home/ehoffer/Datasets/ImageNet/LMDB/validation/', --Validation LMDB location
TRAINING_DIR = '/home/ehoffer/Datasets/ImageNet/LMDB/train/', --Training LMDB location
ImageMinSide = 256, --Minimum side length of saved images
ValidationLabels = torch.load('./ValidationLabels'),
ImageNetClasses = ImageNetClasses,
Normalization = {'simple', 118.380948, 61.896913}, --Default normalization -global mean, std
Compressed = true,
Key = Key
}