-
Notifications
You must be signed in to change notification settings - Fork 3
/
config.yml
267 lines (267 loc) · 5.01 KB
/
config.yml
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
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
AMP_VERBOSE: false
DATALOADER:
ASPECT_RATIO_GROUPING: true
NUM_WORKERS: 1
SIZE_DIVISIBILITY: 32
DATASETS:
TEST:
- coco_2014_val
TRAIN:
- coco_2014_train
DTYPE: float32
INPUT:
BRIGHTNESS: 0.0
CONTRAST: 0.0
HORIZONTAL_FLIP_PROB_TRAIN: 0.5
HUE: 0.0
MAX_SIZE_TEST: 200
MAX_SIZE_TRAIN: 200
MIN_SIZE_TEST: 200
MIN_SIZE_TRAIN:
- 200
PIXEL_MEAN:
- 102.9801
- 115.9465
- 122.7717
PIXEL_STD:
- 1.0
- 1.0
- 1.0
SATURATION: 0.0
TO_BGR255: true
VERTICAL_FLIP_PROB_TRAIN: 0.0
MODEL:
BACKBONE:
CONV_BODY: R-50-FPN
FREEZE_CONV_BODY_AT: 2
CLS_AGNOSTIC_BBOX_REG: false
DEVICE: cuda
FBNET:
ARCH: default
ARCH_DEF: ''
BN_TYPE: bn
DET_HEAD_BLOCKS: []
DET_HEAD_LAST_SCALE: 1.0
DET_HEAD_STRIDE: 0
DW_CONV_SKIP_BN: true
DW_CONV_SKIP_RELU: true
KPTS_HEAD_BLOCKS: []
KPTS_HEAD_LAST_SCALE: 0.0
KPTS_HEAD_STRIDE: 0
MASK_HEAD_BLOCKS: []
MASK_HEAD_LAST_SCALE: 0.0
MASK_HEAD_STRIDE: 0
RPN_BN_TYPE: ''
RPN_HEAD_BLOCKS: 0
SCALE_FACTOR: 1.0
WIDTH_DIVISOR: 1
FPN:
USE_GN: false
USE_RELU: false
GROUP_NORM:
DIM_PER_GP: -1
EPSILON: 1.0e-05
NUM_GROUPS: 32
KEYPOINT_ON: false
MASK_ON: false
META_ARCHITECTURE: GeneralizedRCNN
RESNETS:
BACKBONE_OUT_CHANNELS: 256
DEFORMABLE_GROUPS: 1
NUM_GROUPS: 1
RES2_OUT_CHANNELS: 256
RES5_DILATION: 1
STAGE_WITH_DCN:
- false
- false
- false
- false
STEM_FUNC: StemWithFixedBatchNorm
STEM_OUT_CHANNELS: 64
STRIDE_IN_1X1: true
TRANS_FUNC: BottleneckWithFixedBatchNorm
WIDTH_PER_GROUP: 64
WITH_MODULATED_DCN: false
RETINANET:
ANCHOR_SIZES:
- 4
- 8
- 16
- 32
- 64
ANCHOR_STRIDES:
- 4
- 8
- 16
- 32
- 64
ASPECT_RATIOS:
- 0.1
- 0.2
- 0.3333333333333333
- 0.5
- 1.0
- 2.0
- 3.0
- 4.0
- 5.0
BBOX_REG_BETA: 0.11
BBOX_REG_WEIGHT: 4.0
BG_IOU_THRESHOLD: 0.4
FG_IOU_THRESHOLD: 0.5
INFERENCE_TH: 0.05
LOSS_ALPHA: 0.25
LOSS_GAMMA: 2.0
NMS_TH: 0.4
NUM_CLASSES: 7
NUM_CONVS: 4
OCTAVE: 2.0
PRE_NMS_TOP_N: 1000
PRIOR_PROB: 0.01
SCALES_PER_OCTAVE: 3
STRADDLE_THRESH: 0
USE_C5: true
RETINANET_ON: false
ROI_BOX_HEAD:
CONV_HEAD_DIM: 256
DILATION: 1
FEATURE_EXTRACTOR: FPN2MLPFeatureExtractor
MLP_HEAD_DIM: 1024
NUM_CLASSES: 7
NUM_STACKED_CONVS: 4
POOLER_RESOLUTION: 7
POOLER_SAMPLING_RATIO: 2
POOLER_SCALES:
- 0.25
- 0.125
- 0.0625
- 0.03125
PREDICTOR: FPNPredictor
USE_GN: false
ROI_HEADS:
BATCH_SIZE_PER_IMAGE: 512
BBOX_REG_WEIGHTS:
- 10.0
- 10.0
- 5.0
- 5.0
BG_IOU_THRESHOLD: 0.5
DETECTIONS_PER_IMG: 100
FG_IOU_THRESHOLD: 0.5
NMS: 0.5
POSITIVE_FRACTION: 0.25
SCORE_THRESH: 0.05
USE_FPN: true
ROI_KEYPOINT_HEAD:
CONV_LAYERS:
- 512
- 512
- 512
- 512
- 512
- 512
- 512
- 512
FEATURE_EXTRACTOR: KeypointRCNNFeatureExtractor
MLP_HEAD_DIM: 1024
NUM_CLASSES: 7
POOLER_RESOLUTION: 14
POOLER_SAMPLING_RATIO: 0
POOLER_SCALES:
- 0.0625
PREDICTOR: KeypointRCNNPredictor
RESOLUTION: 14
SHARE_BOX_FEATURE_EXTRACTOR: true
ROI_MASK_HEAD:
CONV_LAYERS:
- 256
- 256
- 256
- 256
DILATION: 1
FEATURE_EXTRACTOR: MaskRCNNFPNFeatureExtractor
MLP_HEAD_DIM: 1024
POOLER_RESOLUTION: 14
POOLER_SAMPLING_RATIO: 2
POOLER_SCALES:
- 0.25
- 0.125
- 0.0625
- 0.03125
POSTPROCESS_MASKS: false
POSTPROCESS_MASKS_THRESHOLD: 0.5
PREDICTOR: MaskRCNNC4Predictor
RESOLUTION: 28
SHARE_BOX_FEATURE_EXTRACTOR: false
USE_GN: false
RPN:
ANCHOR_SIZES:
- 4
- 8
- 16
- 32
- 64
ANCHOR_STRIDE:
- 2
- 4
- 8
- 16
- 32
ASPECT_RATIOS:
- 0.1
- 0.2
- 0.3333333333333333
- 0.5
- 1.0
- 2.0
- 3.0
- 4.0
- 5.0
BATCH_SIZE_PER_IMAGE: 256
BG_IOU_THRESHOLD: 0.3
FG_IOU_THRESHOLD: 0.7
FPN_POST_NMS_PER_BATCH: true
FPN_POST_NMS_TOP_N_TEST: 1000
FPN_POST_NMS_TOP_N_TRAIN: 2000
MIN_SIZE: 0
NMS_THRESH: 0.7
POSITIVE_FRACTION: 0.5
POST_NMS_TOP_N_TEST: 1000
POST_NMS_TOP_N_TRAIN: 2000
PRE_NMS_TOP_N_TEST: 1000
PRE_NMS_TOP_N_TRAIN: 2000
RPN_HEAD: SingleConvRPNHead
STRADDLE_THRESH: 0
USE_FPN: true
RPN_ONLY: false
WEIGHT: /home/heqing/maskrcnn-benchmark/demo/R-50.pkl
OUTPUT_DIR: .
PATHS_CATALOG: /home/heqing/maskrcnn-benchmark/maskrcnn_benchmark/config/paths_catalog.py
SOLVER:
BASE_LR: 0.01
BIAS_LR_FACTOR: 2
CHECKPOINT_PERIOD: 2500
GAMMA: 0.1
IMS_PER_BATCH: 8
MAX_ITER: 50000
MOMENTUM: 0.9
STEPS:
- 6000
- 8000
TEST_PERIOD: 0
WARMUP_FACTOR: 0.3333333333333333
WARMUP_ITERS: 500
WARMUP_METHOD: linear
WEIGHT_DECAY: 0.0001
WEIGHT_DECAY_BIAS: 0
TEST:
BBOX_AUG:
ENABLED: false
H_FLIP: false
MAX_SIZE: 4000
SCALES: []
SCALE_H_FLIP: false
DETECTIONS_PER_IMG: 100
EXPECTED_RESULTS: []
EXPECTED_RESULTS_SIGMA_TOL: 4
IMS_PER_BATCH: 8