-
Notifications
You must be signed in to change notification settings - Fork 1
/
cfg.json
75 lines (75 loc) · 2.34 KB
/
cfg.json
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
{
"Datasets":
{
"base_dir": "./Datasets/",
"DeepGlobe":
{
"train_dir": "/Datasets/DeepGlobe/cropped_train/",
"train_label_dir": "/Datasets/DeepGlobe/cropped_train_labels/",
"valid_dir": "/Datasets/DeepGlobe/cropped_valid/",
"valid_label_dir": "/Datasets/DeepGlobe/cropped_valid_labels/",
"test_dir": "/Datasets/DeepGlobe/cropped_test/",
"hold_dir": "/Datasets/DeepGlobe/cropped_hold/",
"mean" : "[70.95016901, 71.16398124, 71.30953645]",
"std" : "[34.00087859, 35.18201658, 36.40463264]"
},
"MassachusettsRoads":
{
"train_dir": "/Datasets/MassachusettsRoads/cropped_train/",
"train_label_dir": "/Datasets/MassachusettsRoads/cropped_train_labels/",
"valid_dir": "/Datasets/MassachusettsRoads/cropped_valid/",
"valid_label_dir": "/Datasets/MassachusettsRoads/cropped_valid_labels/",
"test_dir": "/Datasets/MassachusettsRoads/cropped_test/",
"test_label_dir": "/Datasets/MassachusettsRoads/cropped_test_labels/",
"mean" : "[70.95016901, 71.16398124, 71.30953645]",
"std" : "[34.00087859, 35.18201658, 36.40463264]"
},
"Spacenet":
{
"train_dir": "/Datasets/Spacenet/cropped_train/",
"train_label_dir": "/Datasets/Spacenet/cropped_train_labels/",
"valid_dir": "/Datasets/Spacenet/cropped_valid/",
"valid_label_dir": "/Datasets/Spacenet/cropped_valid_labels/",
"test_dir": "/Datasets/Spacenet/cropped_test/",
"mean" : "[70.95016901, 71.16398124, 71.30953645]",
"std" : "[34.00087859, 35.18201658, 36.40463264]"
}
},
"Models":
{
"base_dir": "./Models/",
"scales" : "[4, 2, 1]",
"smooth" : "[1, 2, 4]"
},
"training_settings":
{
"batch_size": 11,
"crop_size" : 256,
"epochs": 120,
"results_directory": "./Experiments/",
"augment": true,
"iteration_size": 1,
"roadclass" : 2,
"orientationclass" : 37,
"road_weight": 1,
"road_binary_thresh": 0.76,
"orientation_weight": 1,
"iteration_frequency": 1
},
"validation_settings":
{
"batch_size": 4,
"crop_size" : 512,
"augment": false,
"spacenet_crop_size" : 650,
"evaluation_frequency": 4
},
"optimizer_settings":
{
"learning_rate": 0.01,
"learning_rate_decay": 0.0001,
"learning_rate_step": 0.1,
"learning_rate_drop_at_epoch": "[50,90,110]"
},
"GlobalSeed" : 42
}