-
Notifications
You must be signed in to change notification settings - Fork 2
/
kitti_squeezeDetPlus_config.json
62 lines (59 loc) · 2.38 KB
/
kitti_squeezeDetPlus_config.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
{
"DATASET_NAME" : "KITTI",
"CLASS_NAMES" : ["car", "pedestrian", "cyclist"],
"CLASSES" : 3,
"DATA_PATH" : "/media/terabyte/projects/datasets/kitti/",
"IMAGE_WIDTH" : 1242,
"IMAGE_HEIGHT" : 375,
"BATCH_SIZE" : 20,
"IS_TRAINING" : true,
"ALREADY_PREPROCESSED" : true,
"EVAL_WITH_TRAIN" : true,
"EVAL_ITERS" : 200,
"NET" : "squeezeDet+",
"OPTIMIZER" : { "TYPE": "MOMENTUM",
"MOMENTUM" : 0.9,
"BETA1" : 0.9,
"BETA2" : 0.999
},
"INIT_ANCHOR_SHAPES" : {"VALUE": [[0.02898551, 0.09866667], [0.29146538, 0.464],
[0.09259259, 0.15733333], [0.13043478, 0.232],
[0.03059581, 0.24], [0.20772947, 0.46133333],
[0.18035427, 0.288], [0.06280193, 0.45333333],
[0.05797101, 0.11466667]],
"METHOD": "CONST"},
"WEIGHT_DECAY" : 0.0001,
"LEARNING_RATE" : 0.01,
"DECAY_STEPS" : 10000,
"MAX_GRAD_NORM" : 1.0,
"MOMENTUM" : 0.9,
"LR_DECAY_FACTOR" : 0.5,
"LOSS_COEF_BBOX" : 5.0,
"LOSS_COEF_CONF_POS" : 75.0,
"LOSS_COEF_CONF_NEG" : 100.0,
"LOSS_COEF_CLASS" : 1.0,
"NMS_THRESH" : 0.4,
"PROB_THRESH" : 0.005,
"TOP_N_DETECTION" : 64,
"DATA_AUGMENTATION" : true,
"LOAD_PRETRAINED_MODEL" : false,
"PRETRAINED_MODEL_PATH" : "/media/terabyte/projects/Thesis/SqueezeNet_imageNet_trained/squeezenet_v1.0_SR_0.750.pkl",
"DRIFT_X" : 150,
"DRIFT_Y" : 100,
"EXCLUDE_HARD_EXAMPLES" : false,
"FREEZE_LAYERS" : {
"conv1": false,
"fire2": false,
"fire3": false,
"fire4": false,
"fire5": false,
"fire6": false,
"fire7": false,
"fire8": false,
"fire9": false,
"fire10": false,
"fire11": false,
"conv12": false
},
"TRAIN_DIR" : "/media/terabyte/projects/Thesis/trainings/kitti_tests"
}