-
Notifications
You must be signed in to change notification settings - Fork 5
/
ssd_vgg300_voc_det.json
executable file
·170 lines (170 loc) · 4.4 KB
/
ssd_vgg300_voc_det.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
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
{
"dataset": "voc",
"task": "det",
"method": "single_shot_detector",
"data": {
"num_classes": 21,
"data_dir": "~/DataSet/VOC07+12_DET",
"image_tool": "cv2",
"input_mode": "BGR",
"keep_difficult": false,
"workers": 8
},
"train": {
"batch_size": 32,
"data_transformer": {
"size_mode": "fix_size",
"input_size": [300, 300],
"align_method": "only_scale"
}
},
"val": {
"batch_size": 32,
"use_07_metric": true,
"data_transformer": {
"size_mode": "fix_size",
"input_size": [300, 300],
"align_method": "only_scale"
}
},
"test": {
"input_size": [300, 300],
"scale_search": [1.0]
},
"train_trans": {
"shuffle_trans_seq": ["random_contrast", "random_hue", "random_saturation", "random_brightness", "random_perm"],
"trans_seq": ["random_hflip", "random_pad", "random_crop"],
"random_saturation": {
"ratio": 0.5,
"lower": 0.5,
"upper": 1.5
},
"random_hue": {
"ratio": 0.5,
"delta": 18
},
"random_contrast": {
"ratio": 0.5,
"lower": 0.5,
"upper": 1.5
},
"random_pad": {
"ratio": 0.6,
"up_scale_range": [1.0, 4.0]
},
"random_brightness": {
"ratio": 0.5,
"shift_value": 32
},
"random_perm": {
"ratio": 0.5
},
"random_hflip": {
"ratio": 0.5,
"swap_pair": []
},
"random_crop":{
"ratio": 1.0,
"method": "det",
"grid":[3, 2],
"center_jitter": 20
}
},
"val_trans": {
"trans_seq": []
},
"normalize": {
"div_value": 1,
"mean_value": [104, 117, 123],
"mean": [104.0, 117.0, 123.0],
"std": [1.0, 1.0, 1.0]
},
"gt": {
"anchor_method": "ssd",
"iou_threshold": 0.5,
"num_anchor_list": [4, 6, 6, 6, 4, 4],
"feature_maps_wh": [[38, 38], [19, 19], [10, 10], [5, 5], [3, 3], [1, 1]],
"cur_anchor_sizes": [30, 60, 111, 162, 213, 264, 315],
"aspect_ratio_list": [[2], [2, 3], [2, 3], [2, 3], [2], [2]]
},
"details": {
"color_list": [[255, 170, 30], [0, 0, 70], [244, 35, 232]],
"name_id_dict": {
"aeroplane": 1, "bicycle": 2, "bird": 3, "boat": 4, "bottle": 5, "bus": 6, "car": 7,
"cat": 8, "chair": 9, "cow": 10, "diningtable": 11, "dog": 12, "horse": 13, "motorbike": 14,
"person": 15, "pottedplant": 16, "sheep": 17, "sofa": 18, "train": 19, "tvmonitor": 20
},
"name_seq": ["aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair",
"cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant",
"sheep", "sofa", "train", "tvmonitor"]
},
"checkpoints": {
"checkpoints_name": "ssd_vgg300_voc_det",
"checkpoints_dir": "checkpoints/det/voc",
"save_iters": 5000
},
"network":{
"backbone": "vgg16",
"model_name": "vgg300_ssd",
"num_feature_list": [512, 1024, 512, 256, 256, 256],
"stride_list": [8, 16, 30, 60, 100, 300],
"head_index_list": [0, 1, 2, 3, 4, 5]
},
"logging": {
"logfile_level": "info",
"stdout_level": "info",
"log_file": "./log/det/ssd_vgg300_voc_det.log",
"log_format": "%(asctime)s %(levelname)-7s %(message)s",
"rewrite": true
},
"lr": {
"metric": "epoch",
"is_warm": true,
"warm": {
"warm_epoch": 6,
"freeze_backbone": false
},
"base_lr": 0.004,
"lr_policy": "multistep",
"step": {
"gamma": 0.1,
"step_size": 500
},
"multistep": {
"gamma": 0.1,
"stepvalue": [156, 195, 234]
}
},
"solver": {
"display_iter": 100,
"test_interval": 5000,
"max_epoch": 235
},
"optim": {
"optim_method": "sgd",
"adam": {
"betas": [0.9, 0.999],
"eps": 1e-08,
"weight_decay": 0.0001
},
"sgd":{
"weight_decay": 0.0005,
"momentum": 0.9,
"nesterov": false
}
},
"nms": {
"mode": "union",
"max_threshold": 0.45,
"pre_nms": 1000
},
"res": {
"val_conf_thre": 0.01,
"vis_conf_thre": 0.5,
"max_per_image": 200,
"cls_keep_num": 50
},
"loss": {
"loss_type": "ssd_multibox_loss"
}
}