-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.yml
255 lines (255 loc) · 4.93 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
AMP_VERBOSE: false
DATALOADER:
ASPECT_RATIO_GROUPING: true
NUM_WORKERS: 4
SIZE_DIVISIBILITY: 32
DATASETS:
TEST:
- coco_fish_train
TRAIN:
- coco_fish_train
DTYPE: float32
INPUT:
BRIGHTNESS: 0.0
CONTRAST: 0.0
HORIZONTAL_FLIP_PROB_TRAIN: 0.5
HUE: 0.0
MAX_SIZE_TEST: 1333
MAX_SIZE_TRAIN: 1333
MIN_SIZE_TEST: 800
MIN_SIZE_TRAIN:
- 800
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: true
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:
- 32
- 64
- 128
- 256
- 512
ANCHOR_STRIDES:
- 8
- 16
- 32
- 64
- 128
ASPECT_RATIOS:
- 0.5
- 1.0
- 2.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: 81
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: 81
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: 17
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:
- 32
- 64
- 128
- 256
- 512
ANCHOR_STRIDE:
- 4
- 8
- 16
- 32
- 64
ASPECT_RATIOS:
- 0.5
- 1.0
- 2.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: catalog://ImageNetPretrained/MSRA/R-50
OUTPUT_DIR: .
PATHS_CATALOG: /usr/local/lib/python3.6/dist-packages/maskrcnn_benchmark-0.1-py3.6-linux-aarch64.egg/maskrcnn_benchmark/config/paths_catalog.py
SOLVER:
BASE_LR: 0.0025
BIAS_LR_FACTOR: 2
CHECKPOINT_PERIOD: 2500
GAMMA: 0.1
IMS_PER_BATCH: 1
MAX_ITER: 200
MOMENTUM: 0.9
STEPS:
- 480000
- 640000
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: 1