Hugging Face's logo

nabeelkhalid92
/
wsbench-models

detectron2
cell-segmentation
instance-segmentation
weak-supervision
microscopy
mask-rcnn
Model card Files
xet
 Community
nabeelkhalid92 commited on
Commit
89a3350
·
verified ·
1 Parent(s): 936cae9

Add files using upload-large-folder tool

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. dsb2018/mrcnn_boxrect/box/seed0/config.yaml +324 -0
  2. dsb2018/mrcnn_boxrect/box/seed0/run_info.json +23 -0
  3. dsb2018/mrcnn_boxrect/box/seed0/score_thr.json +52 -0
  4. dsb2018/mrcnn_boxrect/box/seed0/val_history.json +115 -0
  5. dsb2018/mrcnn_boxrect/box/seed1/config.yaml +324 -0
  6. dsb2018/mrcnn_boxrect/box/seed1/run_info.json +25 -0
  7. dsb2018/mrcnn_boxrect/box/seed1/score_thr.json +52 -0
  8. dsb2018/mrcnn_boxrect/box/seed1/val_history.json +115 -0
  9. dsb2018/mrcnn_boxrect/box/seed2/config.yaml +324 -0
  10. dsb2018/mrcnn_boxrect/box/seed2/run_info.json +25 -0
  11. dsb2018/mrcnn_boxrect/box/seed2/score_thr.json +52 -0
  12. dsb2018/mrcnn_boxrect/box/seed2/val_history.json +115 -0
  13. dsb2018/mrcnn_full/budget_mask_B10288/seed0/config.yaml +324 -0
  14. dsb2018/mrcnn_full/budget_mask_B10288/seed0/run_info.json +25 -0
  15. dsb2018/mrcnn_full/budget_mask_B10288/seed0/score_thr.json +52 -0
  16. dsb2018/mrcnn_full/budget_mask_B10288/seed0/val_history.json +115 -0
  17. dsb2018/mrcnn_full/budget_mask_B10288/seed1/config.yaml +324 -0
  18. dsb2018/mrcnn_full/budget_mask_B10288/seed1/run_info.json +25 -0
  19. dsb2018/mrcnn_full/budget_mask_B10288/seed1/score_thr.json +52 -0
  20. dsb2018/mrcnn_full/budget_mask_B10288/seed1/val_history.json +115 -0
  21. dsb2018/mrcnn_full/budget_mask_B10288/seed2/config.yaml +324 -0
  22. dsb2018/mrcnn_full/budget_mask_B10288/seed2/run_info.json +25 -0
  23. dsb2018/mrcnn_full/budget_mask_B10288/seed2/score_thr.json +52 -0
  24. dsb2018/mrcnn_full/budget_mask_B10288/seed2/val_history.json +115 -0
  25. dsb2018/mrcnn_full/budget_mask_B10288/seed3/config.yaml +324 -0
  26. dsb2018/mrcnn_full/budget_mask_B10288/seed3/run_info.json +25 -0
  27. dsb2018/mrcnn_full/budget_mask_B10288/seed3/score_thr.json +52 -0
  28. dsb2018/mrcnn_full/budget_mask_B10288/seed3/val_history.json +115 -0
  29. dsb2018/mrcnn_full/budget_mask_B10288/seed4/config.yaml +324 -0
  30. dsb2018/mrcnn_full/budget_mask_B10288/seed4/run_info.json +25 -0
  31. dsb2018/mrcnn_full/budget_mask_B10288/seed4/score_thr.json +52 -0
  32. dsb2018/mrcnn_full/budget_mask_B10288/seed4/val_history.json +115 -0
  33. dsb2018/mrcnn_full/budget_mask_B164605/seed0/config.yaml +324 -0
  34. dsb2018/mrcnn_full/budget_mask_B164605/seed0/run_info.json +25 -0
  35. dsb2018/mrcnn_full/budget_mask_B164605/seed0/score_thr.json +52 -0
  36. dsb2018/mrcnn_full/budget_mask_B164605/seed0/val_history.json +115 -0
  37. dsb2018/mrcnn_full/budget_mask_B164605/seed1/config.yaml +324 -0
  38. dsb2018/mrcnn_full/budget_mask_B164605/seed1/run_info.json +25 -0
  39. dsb2018/mrcnn_full/budget_mask_B164605/seed1/score_thr.json +52 -0
  40. dsb2018/mrcnn_full/budget_mask_B164605/seed1/val_history.json +115 -0
  41. dsb2018/mrcnn_full/budget_mask_B164605/seed2/config.yaml +324 -0
  42. dsb2018/mrcnn_full/budget_mask_B164605/seed2/run_info.json +25 -0
  43. dsb2018/mrcnn_full/budget_mask_B164605/seed2/score_thr.json +52 -0
  44. dsb2018/mrcnn_full/budget_mask_B164605/seed2/val_history.json +115 -0
  45. dsb2018/mrcnn_full/budget_mask_B164605/seed3/config.yaml +324 -0
  46. dsb2018/mrcnn_full/budget_mask_B164605/seed3/run_info.json +25 -0
  47. dsb2018/mrcnn_full/budget_mask_B164605/seed3/score_thr.json +52 -0
  48. dsb2018/mrcnn_full/budget_mask_B164605/seed3/val_history.json +115 -0
  49. dsb2018/mrcnn_full/budget_mask_B164605/seed4/config.yaml +324 -0
  50. dsb2018/mrcnn_full/budget_mask_B164605/seed4/run_info.json +25 -0
dsb2018/mrcnn_boxrect/box/seed0/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__train_box
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_boxrect/box/seed0
273
+ SEED: 0
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_boxrect/box/seed0/run_info.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_boxrect",
4
+ "setting": "box",
5
+ "seed": 0,
6
+ "train_json": "labels/dsb2018/train_box.json",
7
+ "gpu": "Tesla V100-SXM2-32GB",
8
+ "torch": "1.10.2",
9
+ "train_sec": 4225.9,
10
+ "best_iter": 3500,
11
+ "best_val": 39.9026934790469,
12
+ "select_metric": "AP",
13
+ "predict": {
14
+ "val": {
15
+ "n_dets": 5170,
16
+ "sec": 8.1
17
+ },
18
+ "test": {
19
+ "n_dets": 5616,
20
+ "sec": 13.8
21
+ }
22
+ }
23
+ }
dsb2018/mrcnn_boxrect/box/seed0/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 60.90657604938523,
9
+ "F1@0.5": 83.71570672228461
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 61.44342394947457,
14
+ "F1@0.5": 84.43579766536965
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 61.787987485170135,
19
+ "F1@0.5": 84.91674828599413
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 62.22213254197346,
24
+ "F1@0.5": 85.50247116968698
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 62.469185469723485,
29
+ "F1@0.5": 85.79488881513441
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 62.78586011170443,
34
+ "F1@0.5": 86.21553884711778
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 63.02572216541168,
39
+ "F1@0.5": 86.51288095638996
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 63.412545357087765,
44
+ "F1@0.5": 87.00680272108843
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 63.67832665174735,
49
+ "F1@0.5": 87.33699382292382
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_boxrect/box/seed0/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 39.9026934790469,
4
+ "best_iter": 3500,
5
+ "history": [
6
+ {
7
+ "AP": 32.024715599268696,
8
+ "AP50": 84.30324602411278,
9
+ "AP75": 8.067366801413577,
10
+ "AP_cat1": 32.024715599268696,
11
+ "iter": 500,
12
+ "eval_sec": 8.5,
13
+ "n_dets": 6412
14
+ },
15
+ {
16
+ "AP": 30.63309682961491,
17
+ "AP50": 84.354948821064,
18
+ "AP75": 7.123992468953958,
19
+ "AP_cat1": 30.63309682961491,
20
+ "iter": 1000,
21
+ "eval_sec": 9.2,
22
+ "n_dets": 6078
23
+ },
24
+ {
25
+ "AP": 36.91080496028785,
26
+ "AP50": 88.65100298750862,
27
+ "AP75": 15.5713231017222,
28
+ "AP_cat1": 36.91080496028785,
29
+ "iter": 1500,
30
+ "eval_sec": 9.7,
31
+ "n_dets": 6495
32
+ },
33
+ {
34
+ "AP": 37.106265644112106,
35
+ "AP50": 86.82022985800549,
36
+ "AP75": 19.561827534780367,
37
+ "AP_cat1": 37.106265644112106,
38
+ "iter": 2000,
39
+ "eval_sec": 8.5,
40
+ "n_dets": 5086
41
+ },
42
+ {
43
+ "AP": 37.86940808960258,
44
+ "AP50": 88.86159784235899,
45
+ "AP75": 18.340209924611614,
46
+ "AP_cat1": 37.86940808960258,
47
+ "iter": 2500,
48
+ "eval_sec": 8.4,
49
+ "n_dets": 5577
50
+ },
51
+ {
52
+ "AP": 38.85034834161433,
53
+ "AP50": 90.11639722475802,
54
+ "AP75": 18.786390842484444,
55
+ "AP_cat1": 38.85034834161433,
56
+ "iter": 3000,
57
+ "eval_sec": 8.2,
58
+ "n_dets": 6081
59
+ },
60
+ {
61
+ "AP": 39.9026934790469,
62
+ "AP50": 89.7201574339429,
63
+ "AP75": 23.14803526371244,
64
+ "AP_cat1": 39.9026934790469,
65
+ "iter": 3500,
66
+ "eval_sec": 9.1,
67
+ "n_dets": 5170
68
+ },
69
+ {
70
+ "AP": 35.88658821533334,
71
+ "AP50": 87.69364655655636,
72
+ "AP75": 15.087402794176313,
73
+ "AP_cat1": 35.88658821533334,
74
+ "iter": 4000,
75
+ "eval_sec": 8.8,
76
+ "n_dets": 5404
77
+ },
78
+ {
79
+ "AP": 35.23806444880416,
80
+ "AP50": 87.91260364673153,
81
+ "AP75": 13.075030813243371,
82
+ "AP_cat1": 35.23806444880416,
83
+ "iter": 4500,
84
+ "eval_sec": 8.8,
85
+ "n_dets": 4711
86
+ },
87
+ {
88
+ "AP": 38.16296163795442,
89
+ "AP50": 89.44472691366538,
90
+ "AP75": 19.36519891707676,
91
+ "AP_cat1": 38.16296163795442,
92
+ "iter": 5000,
93
+ "eval_sec": 10.5,
94
+ "n_dets": 4761
95
+ },
96
+ {
97
+ "AP": 37.71881829036722,
98
+ "AP50": 89.54552388384272,
99
+ "AP75": 17.95055580866726,
100
+ "AP_cat1": 37.71881829036722,
101
+ "iter": 5500,
102
+ "eval_sec": 7.4,
103
+ "n_dets": 4784
104
+ },
105
+ {
106
+ "AP": 37.844239642667645,
107
+ "AP50": 89.49763196661313,
108
+ "AP75": 18.64205334706184,
109
+ "AP_cat1": 37.844239642667645,
110
+ "iter": 6000,
111
+ "eval_sec": 8.9,
112
+ "n_dets": 4779
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_boxrect/box/seed1/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__train_box
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_boxrect/box/seed1
273
+ SEED: 1
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_boxrect/box/seed1/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_boxrect",
4
+ "setting": "box",
5
+ "seed": 1,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/train_box.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 4404.6,
12
+ "best_iter": 500,
13
+ "best_val": 39.02773676302599,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 6765,
18
+ "sec": 9.9
19
+ },
20
+ "test": {
21
+ "n_dets": 7201,
22
+ "sec": 18.8
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_boxrect/box/seed1/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 57.54449868370536,
9
+ "F1@0.5": 79.00763358778626
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 58.38405580724793,
14
+ "F1@0.5": 80.14860281053143
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 59.08204870119673,
19
+ "F1@0.5": 81.06955380577428
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 59.74635490519961,
24
+ "F1@0.5": 81.96339434276206
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 60.26407521539615,
29
+ "F1@0.5": 82.62989742727426
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 60.672755846432345,
34
+ "F1@0.5": 83.15180394826413
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 61.07553309656843,
39
+ "F1@0.5": 83.66574965612105
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 61.264858223100426,
44
+ "F1@0.5": 83.80257212374
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 61.590814600531886,
49
+ "F1@0.5": 84.17164966584593
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_boxrect/box/seed1/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 39.02773676302599,
4
+ "best_iter": 500,
5
+ "history": [
6
+ {
7
+ "AP": 39.02773676302599,
8
+ "AP50": 87.00814549116114,
9
+ "AP75": 22.864660901438512,
10
+ "AP_cat1": 39.02773676302599,
11
+ "iter": 500,
12
+ "eval_sec": 10.8,
13
+ "n_dets": 6765
14
+ },
15
+ {
16
+ "AP": 34.20685865527156,
17
+ "AP50": 86.7937462140749,
18
+ "AP75": 10.386961205313956,
19
+ "AP_cat1": 34.20685865527156,
20
+ "iter": 1000,
21
+ "eval_sec": 10.5,
22
+ "n_dets": 6544
23
+ },
24
+ {
25
+ "AP": 37.42621155622322,
26
+ "AP50": 88.14388143363435,
27
+ "AP75": 18.29274610999254,
28
+ "AP_cat1": 37.42621155622322,
29
+ "iter": 1500,
30
+ "eval_sec": 9.2,
31
+ "n_dets": 6329
32
+ },
33
+ {
34
+ "AP": 29.072479263153046,
35
+ "AP50": 85.34504504656266,
36
+ "AP75": 4.269163065903734,
37
+ "AP_cat1": 29.072479263153046,
38
+ "iter": 2000,
39
+ "eval_sec": 10.6,
40
+ "n_dets": 6262
41
+ },
42
+ {
43
+ "AP": 36.61519504200092,
44
+ "AP50": 88.20910839966042,
45
+ "AP75": 16.321240493656493,
46
+ "AP_cat1": 36.61519504200092,
47
+ "iter": 2500,
48
+ "eval_sec": 10.0,
49
+ "n_dets": 6246
50
+ },
51
+ {
52
+ "AP": 36.67758814870045,
53
+ "AP50": 88.2763570320482,
54
+ "AP75": 15.706495232978558,
55
+ "AP_cat1": 36.67758814870045,
56
+ "iter": 3000,
57
+ "eval_sec": 9.1,
58
+ "n_dets": 4852
59
+ },
60
+ {
61
+ "AP": 36.946665790564545,
62
+ "AP50": 88.63007875196234,
63
+ "AP75": 16.269340804506353,
64
+ "AP_cat1": 36.946665790564545,
65
+ "iter": 3500,
66
+ "eval_sec": 9.1,
67
+ "n_dets": 5377
68
+ },
69
+ {
70
+ "AP": 38.290436996845884,
71
+ "AP50": 89.31914080352014,
72
+ "AP75": 18.90728025557018,
73
+ "AP_cat1": 38.290436996845884,
74
+ "iter": 4000,
75
+ "eval_sec": 7.5,
76
+ "n_dets": 5083
77
+ },
78
+ {
79
+ "AP": 36.39705569194051,
80
+ "AP50": 88.27157179159983,
81
+ "AP75": 15.73868255762298,
82
+ "AP_cat1": 36.39705569194051,
83
+ "iter": 4500,
84
+ "eval_sec": 8.4,
85
+ "n_dets": 4631
86
+ },
87
+ {
88
+ "AP": 36.585440405735895,
89
+ "AP50": 88.61151690880153,
90
+ "AP75": 15.791678578666634,
91
+ "AP_cat1": 36.585440405735895,
92
+ "iter": 5000,
93
+ "eval_sec": 7.5,
94
+ "n_dets": 4684
95
+ },
96
+ {
97
+ "AP": 37.757351069723704,
98
+ "AP50": 89.54660460626296,
99
+ "AP75": 18.424450091540436,
100
+ "AP_cat1": 37.757351069723704,
101
+ "iter": 5500,
102
+ "eval_sec": 8.5,
103
+ "n_dets": 4683
104
+ },
105
+ {
106
+ "AP": 37.368386716253,
107
+ "AP50": 89.41412128055704,
108
+ "AP75": 17.49035613399853,
109
+ "AP_cat1": 37.368386716253,
110
+ "iter": 6000,
111
+ "eval_sec": 8.7,
112
+ "n_dets": 4708
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_boxrect/box/seed2/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__train_box
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_boxrect/box/seed2
273
+ SEED: 2
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_boxrect/box/seed2/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_boxrect",
4
+ "setting": "box",
5
+ "seed": 2,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/train_box.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 5643.6,
12
+ "best_iter": 4000,
13
+ "best_val": 41.061559751396324,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 5564,
18
+ "sec": 8.9
19
+ },
20
+ "test": {
21
+ "n_dets": 5523,
22
+ "sec": 12.5
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_boxrect/box/seed2/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 60.86565169850099,
9
+ "F1@0.5": 83.13253012048193
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 61.43820236703202,
14
+ "F1@0.5": 83.91541172701056
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 62.10589625178909,
19
+ "F1@0.5": 84.8199805384366
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 62.50024772743256,
24
+ "F1@0.5": 85.35469107551488
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 62.97341571781435,
29
+ "F1@0.5": 85.98315998018822
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 63.37857329319643,
34
+ "F1@0.5": 86.5125561283885
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 63.640720534500474,
39
+ "F1@0.5": 86.8588669125042
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 63.869599257895494,
44
+ "F1@0.5": 87.12837837837839
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 64.16471599688293,
49
+ "F1@0.5": 87.46803069053708
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_boxrect/box/seed2/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 41.061559751396324,
4
+ "best_iter": 4000,
5
+ "history": [
6
+ {
7
+ "AP": 34.37825697298146,
8
+ "AP50": 85.57867654158773,
9
+ "AP75": 13.816471769731745,
10
+ "AP_cat1": 34.37825697298146,
11
+ "iter": 500,
12
+ "eval_sec": 8.3,
13
+ "n_dets": 6949
14
+ },
15
+ {
16
+ "AP": 37.45255738219518,
17
+ "AP50": 86.42891720954802,
18
+ "AP75": 19.760843057055357,
19
+ "AP_cat1": 37.45255738219518,
20
+ "iter": 1000,
21
+ "eval_sec": 9.8,
22
+ "n_dets": 5930
23
+ },
24
+ {
25
+ "AP": 40.319245332819,
26
+ "AP50": 88.99237413721565,
27
+ "AP75": 25.238223005270566,
28
+ "AP_cat1": 40.319245332819,
29
+ "iter": 1500,
30
+ "eval_sec": 10.3,
31
+ "n_dets": 5935
32
+ },
33
+ {
34
+ "AP": 37.31276274875306,
35
+ "AP50": 88.97910212959484,
36
+ "AP75": 17.006814173547774,
37
+ "AP_cat1": 37.31276274875306,
38
+ "iter": 2000,
39
+ "eval_sec": 12.8,
40
+ "n_dets": 6553
41
+ },
42
+ {
43
+ "AP": 33.2785439275159,
44
+ "AP50": 88.2973923474655,
45
+ "AP75": 8.48736824285348,
46
+ "AP_cat1": 33.2785439275159,
47
+ "iter": 2500,
48
+ "eval_sec": 12.2,
49
+ "n_dets": 5816
50
+ },
51
+ {
52
+ "AP": 35.60599039777954,
53
+ "AP50": 87.75308595237048,
54
+ "AP75": 13.761117533835709,
55
+ "AP_cat1": 35.60599039777954,
56
+ "iter": 3000,
57
+ "eval_sec": 8.5,
58
+ "n_dets": 5664
59
+ },
60
+ {
61
+ "AP": 36.070618878576475,
62
+ "AP50": 88.52167105088039,
63
+ "AP75": 12.865421237665794,
64
+ "AP_cat1": 36.070618878576475,
65
+ "iter": 3500,
66
+ "eval_sec": 9.7,
67
+ "n_dets": 5513
68
+ },
69
+ {
70
+ "AP": 41.061559751396324,
71
+ "AP50": 90.22435588140458,
72
+ "AP75": 26.876576475649017,
73
+ "AP_cat1": 41.061559751396324,
74
+ "iter": 4000,
75
+ "eval_sec": 8.3,
76
+ "n_dets": 5564
77
+ },
78
+ {
79
+ "AP": 32.22025983504071,
80
+ "AP50": 86.97369529883204,
81
+ "AP75": 7.850199837537524,
82
+ "AP_cat1": 32.22025983504071,
83
+ "iter": 4500,
84
+ "eval_sec": 7.7,
85
+ "n_dets": 5127
86
+ },
87
+ {
88
+ "AP": 38.07805462619282,
89
+ "AP50": 89.29890917361793,
90
+ "AP75": 19.136341724290197,
91
+ "AP_cat1": 38.07805462619282,
92
+ "iter": 5000,
93
+ "eval_sec": 11.8,
94
+ "n_dets": 4799
95
+ },
96
+ {
97
+ "AP": 37.126125465320584,
98
+ "AP50": 88.43950483030365,
99
+ "AP75": 16.97621217819322,
100
+ "AP_cat1": 37.126125465320584,
101
+ "iter": 5500,
102
+ "eval_sec": 10.8,
103
+ "n_dets": 4773
104
+ },
105
+ {
106
+ "AP": 37.36006433666258,
107
+ "AP50": 88.55372331110358,
108
+ "AP75": 17.18987521345769,
109
+ "AP_cat1": 37.36006433666258,
110
+ "iter": 6000,
111
+ "eval_sec": 10.5,
112
+ "n_dets": 4756
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed0/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__mask_B10288_s0
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_full/budget_mask_B10288/seed0
273
+ SEED: 0
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_full/budget_mask_B10288/seed0/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_full",
4
+ "setting": "budget_mask_B10288",
5
+ "seed": 0,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/budget/mask_B10288_s0.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 3342.3,
12
+ "best_iter": 2000,
13
+ "best_val": 55.15395073636342,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 3057,
18
+ "sec": 4.5
19
+ },
20
+ "test": {
21
+ "n_dets": 3016,
22
+ "sec": 5.6
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed0/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 71.20357571385422,
9
+ "F1@0.5": 85.54381179679508
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 71.3609014106755,
14
+ "F1@0.5": 85.7289352247479
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 71.39889650954548,
19
+ "F1@0.5": 85.74854402192533
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 71.37743054411568,
24
+ "F1@0.5": 85.67994505494505
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 71.43805786069792,
29
+ "F1@0.5": 85.69951757408684
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 71.49539843643076,
34
+ "F1@0.5": 85.74390876101607
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 71.55906591730005,
39
+ "F1@0.5": 85.79348579348579
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 71.6456571181671,
44
+ "F1@0.5": 85.85823488533704
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 71.75953203251454,
49
+ "F1@0.5": 85.98293574786697
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed0/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 55.15395073636342,
4
+ "best_iter": 2000,
5
+ "history": [
6
+ {
7
+ "AP": 54.16962124026078,
8
+ "AP50": 83.50640257127328,
9
+ "AP75": 59.99715560749011,
10
+ "AP_cat1": 54.16962124026078,
11
+ "iter": 500,
12
+ "eval_sec": 4.7,
13
+ "n_dets": 3628
14
+ },
15
+ {
16
+ "AP": 54.11339246325907,
17
+ "AP50": 83.3615517942679,
18
+ "AP75": 61.0252781223357,
19
+ "AP_cat1": 54.11339246325907,
20
+ "iter": 1000,
21
+ "eval_sec": 4.7,
22
+ "n_dets": 3197
23
+ },
24
+ {
25
+ "AP": 54.559214434603895,
26
+ "AP50": 83.00806769056476,
27
+ "AP75": 61.40499166378247,
28
+ "AP_cat1": 54.559214434603895,
29
+ "iter": 1500,
30
+ "eval_sec": 4.6,
31
+ "n_dets": 3033
32
+ },
33
+ {
34
+ "AP": 55.15395073636342,
35
+ "AP50": 84.07053051209861,
36
+ "AP75": 62.30267166708823,
37
+ "AP_cat1": 55.15395073636342,
38
+ "iter": 2000,
39
+ "eval_sec": 4.5,
40
+ "n_dets": 3057
41
+ },
42
+ {
43
+ "AP": 54.46088831119709,
44
+ "AP50": 82.90037177700577,
45
+ "AP75": 61.27189468275635,
46
+ "AP_cat1": 54.46088831119709,
47
+ "iter": 2500,
48
+ "eval_sec": 4.5,
49
+ "n_dets": 2987
50
+ },
51
+ {
52
+ "AP": 53.56092838893457,
53
+ "AP50": 81.91902813955856,
54
+ "AP75": 59.88620335262199,
55
+ "AP_cat1": 53.56092838893457,
56
+ "iter": 3000,
57
+ "eval_sec": 4.7,
58
+ "n_dets": 2934
59
+ },
60
+ {
61
+ "AP": 53.93278667156096,
62
+ "AP50": 82.62354718735926,
63
+ "AP75": 60.660274562021,
64
+ "AP_cat1": 53.93278667156096,
65
+ "iter": 3500,
66
+ "eval_sec": 4.6,
67
+ "n_dets": 2963
68
+ },
69
+ {
70
+ "AP": 53.96713812413264,
71
+ "AP50": 82.4729932040706,
72
+ "AP75": 60.40561158287452,
73
+ "AP_cat1": 53.96713812413264,
74
+ "iter": 4000,
75
+ "eval_sec": 4.5,
76
+ "n_dets": 2967
77
+ },
78
+ {
79
+ "AP": 53.77509085719551,
80
+ "AP50": 81.96541546113407,
81
+ "AP75": 60.58947686307539,
82
+ "AP_cat1": 53.77509085719551,
83
+ "iter": 4500,
84
+ "eval_sec": 4.6,
85
+ "n_dets": 2959
86
+ },
87
+ {
88
+ "AP": 54.04321832886934,
89
+ "AP50": 81.93909594302148,
90
+ "AP75": 60.900776783683064,
91
+ "AP_cat1": 54.04321832886934,
92
+ "iter": 5000,
93
+ "eval_sec": 4.6,
94
+ "n_dets": 2930
95
+ },
96
+ {
97
+ "AP": 53.99914421282978,
98
+ "AP50": 81.99967844235051,
99
+ "AP75": 61.00543469058217,
100
+ "AP_cat1": 53.99914421282978,
101
+ "iter": 5500,
102
+ "eval_sec": 4.5,
103
+ "n_dets": 2914
104
+ },
105
+ {
106
+ "AP": 54.120642271281305,
107
+ "AP50": 82.04206106174212,
108
+ "AP75": 61.08986943610315,
109
+ "AP_cat1": 54.120642271281305,
110
+ "iter": 6000,
111
+ "eval_sec": 4.7,
112
+ "n_dets": 2916
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed1/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__mask_B10288_s1
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_full/budget_mask_B10288/seed1
273
+ SEED: 1
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_full/budget_mask_B10288/seed1/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_full",
4
+ "setting": "budget_mask_B10288",
5
+ "seed": 1,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/budget/mask_B10288_s1.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 3679.2,
12
+ "best_iter": 2000,
13
+ "best_val": 39.703357768626695,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 2229,
18
+ "sec": 4.2
19
+ },
20
+ "test": {
21
+ "n_dets": 2384,
22
+ "sec": 4.7
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed1/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 60.04335670040833,
9
+ "F1@0.5": 70.40510876072639
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 60.235687772922155,
14
+ "F1@0.5": 70.63063063063063
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 60.395808027250794,
19
+ "F1@0.5": 70.817433219522
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 60.53106948329286,
24
+ "F1@0.5": 70.9508460918614
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 60.64109354733444,
29
+ "F1@0.5": 71.07070707070707
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 60.67454761569616,
34
+ "F1@0.5": 71.10751164203279
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 60.74932759062992,
39
+ "F1@0.5": 71.18506493506493
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 60.768461212165214,
44
+ "F1@0.5": 71.18161480577588
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 60.84344789759439,
49
+ "F1@0.5": 71.24796084828712
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed1/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 39.703357768626695,
4
+ "best_iter": 2000,
5
+ "history": [
6
+ {
7
+ "AP": 38.70398932149571,
8
+ "AP50": 57.66156445446264,
9
+ "AP75": 45.07397070028802,
10
+ "AP_cat1": 38.70398932149571,
11
+ "iter": 500,
12
+ "eval_sec": 4.4,
13
+ "n_dets": 2549
14
+ },
15
+ {
16
+ "AP": 39.43700609142075,
17
+ "AP50": 57.97990365285634,
18
+ "AP75": 46.155052443242376,
19
+ "AP_cat1": 39.43700609142075,
20
+ "iter": 1000,
21
+ "eval_sec": 4.3,
22
+ "n_dets": 2354
23
+ },
24
+ {
25
+ "AP": 39.62379094841666,
26
+ "AP50": 58.002079506324975,
27
+ "AP75": 46.2915090444647,
28
+ "AP_cat1": 39.62379094841666,
29
+ "iter": 1500,
30
+ "eval_sec": 4.1,
31
+ "n_dets": 2282
32
+ },
33
+ {
34
+ "AP": 39.703357768626695,
35
+ "AP50": 58.84264663768622,
36
+ "AP75": 45.5489673271135,
37
+ "AP_cat1": 39.703357768626695,
38
+ "iter": 2000,
39
+ "eval_sec": 4.1,
40
+ "n_dets": 2229
41
+ },
42
+ {
43
+ "AP": 38.778874237450175,
44
+ "AP50": 57.169666879562996,
45
+ "AP75": 44.870131903728264,
46
+ "AP_cat1": 38.778874237450175,
47
+ "iter": 2500,
48
+ "eval_sec": 4.1,
49
+ "n_dets": 2110
50
+ },
51
+ {
52
+ "AP": 38.560630446353365,
53
+ "AP50": 57.621481568126,
54
+ "AP75": 44.59316682529042,
55
+ "AP_cat1": 38.560630446353365,
56
+ "iter": 3000,
57
+ "eval_sec": 4.2,
58
+ "n_dets": 2159
59
+ },
60
+ {
61
+ "AP": 38.865370606988705,
62
+ "AP50": 57.91874898317556,
63
+ "AP75": 44.549575328164245,
64
+ "AP_cat1": 38.865370606988705,
65
+ "iter": 3500,
66
+ "eval_sec": 4.1,
67
+ "n_dets": 2166
68
+ },
69
+ {
70
+ "AP": 39.52069224757995,
71
+ "AP50": 57.86563044946019,
72
+ "AP75": 46.18526977472829,
73
+ "AP_cat1": 39.52069224757995,
74
+ "iter": 4000,
75
+ "eval_sec": 4.1,
76
+ "n_dets": 2154
77
+ },
78
+ {
79
+ "AP": 38.890111230272225,
80
+ "AP50": 57.917384835613525,
81
+ "AP75": 44.38811453648586,
82
+ "AP_cat1": 38.890111230272225,
83
+ "iter": 4500,
84
+ "eval_sec": 4.2,
85
+ "n_dets": 2155
86
+ },
87
+ {
88
+ "AP": 38.79143135956918,
89
+ "AP50": 57.76812164120615,
90
+ "AP75": 44.68647557325759,
91
+ "AP_cat1": 38.79143135956918,
92
+ "iter": 5000,
93
+ "eval_sec": 4.3,
94
+ "n_dets": 2138
95
+ },
96
+ {
97
+ "AP": 38.617120336176775,
98
+ "AP50": 57.84714754904955,
99
+ "AP75": 44.370087289603525,
100
+ "AP_cat1": 38.617120336176775,
101
+ "iter": 5500,
102
+ "eval_sec": 4.2,
103
+ "n_dets": 2137
104
+ },
105
+ {
106
+ "AP": 38.661802947393674,
107
+ "AP50": 57.786336999218534,
108
+ "AP75": 44.423610994551865,
109
+ "AP_cat1": 38.661802947393674,
110
+ "iter": 6000,
111
+ "eval_sec": 4.4,
112
+ "n_dets": 2145
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed2/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__mask_B10288_s2
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_full/budget_mask_B10288/seed2
273
+ SEED: 2
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_full/budget_mask_B10288/seed2/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_full",
4
+ "setting": "budget_mask_B10288",
5
+ "seed": 2,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/budget/mask_B10288_s2.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 3543.1,
12
+ "best_iter": 500,
13
+ "best_val": 34.864153766672096,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 2977,
18
+ "sec": 8.6
19
+ },
20
+ "test": {
21
+ "n_dets": 3388,
22
+ "sec": 11.3
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed2/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 54.03987545286308,
9
+ "F1@0.5": 66.67946257197697
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 54.510697213827044,
14
+ "F1@0.5": 67.26040658276864
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 54.77745528648322,
19
+ "F1@0.5": 67.58808643176951
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 55.04724019648948,
24
+ "F1@0.5": 67.90752351097179
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 55.332837866815936,
29
+ "F1@0.5": 68.24644549763033
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 55.53328825760276,
34
+ "F1@0.5": 68.48171701112878
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 55.832868434017406,
39
+ "F1@0.5": 68.85114885114885
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 56.06986785981343,
44
+ "F1@0.5": 69.14274242120057
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 56.311583042207204,
49
+ "F1@0.5": 69.43715957232197
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed2/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 34.864153766672096,
4
+ "best_iter": 500,
5
+ "history": [
6
+ {
7
+ "AP": 34.864153766672096,
8
+ "AP50": 58.61967856865916,
9
+ "AP75": 39.01842870570245,
10
+ "AP_cat1": 34.864153766672096,
11
+ "iter": 500,
12
+ "eval_sec": 4.3,
13
+ "n_dets": 2977
14
+ },
15
+ {
16
+ "AP": 33.12625416915461,
17
+ "AP50": 58.21211023118262,
18
+ "AP75": 34.54694051120152,
19
+ "AP_cat1": 33.12625416915461,
20
+ "iter": 1000,
21
+ "eval_sec": 4.1,
22
+ "n_dets": 2507
23
+ },
24
+ {
25
+ "AP": 31.1121895264817,
26
+ "AP50": 56.97992849850573,
27
+ "AP75": 31.663305622653432,
28
+ "AP_cat1": 31.1121895264817,
29
+ "iter": 1500,
30
+ "eval_sec": 4.2,
31
+ "n_dets": 2600
32
+ },
33
+ {
34
+ "AP": 34.07046125127865,
35
+ "AP50": 58.35244410078677,
36
+ "AP75": 35.47809608264208,
37
+ "AP_cat1": 34.07046125127865,
38
+ "iter": 2000,
39
+ "eval_sec": 7.6,
40
+ "n_dets": 2425
41
+ },
42
+ {
43
+ "AP": 33.65770283454853,
44
+ "AP50": 58.838785443556205,
45
+ "AP75": 35.283097051936124,
46
+ "AP_cat1": 33.65770283454853,
47
+ "iter": 2500,
48
+ "eval_sec": 8.9,
49
+ "n_dets": 2411
50
+ },
51
+ {
52
+ "AP": 34.00142147394679,
53
+ "AP50": 58.12778742947491,
54
+ "AP75": 35.44244607312999,
55
+ "AP_cat1": 34.00142147394679,
56
+ "iter": 3000,
57
+ "eval_sec": 7.6,
58
+ "n_dets": 2412
59
+ },
60
+ {
61
+ "AP": 34.798472753825244,
62
+ "AP50": 58.714413946707076,
63
+ "AP75": 36.81526427096647,
64
+ "AP_cat1": 34.798472753825244,
65
+ "iter": 3500,
66
+ "eval_sec": 8.1,
67
+ "n_dets": 2328
68
+ },
69
+ {
70
+ "AP": 33.38568809737693,
71
+ "AP50": 58.30452710283262,
72
+ "AP75": 35.103350508440855,
73
+ "AP_cat1": 33.38568809737693,
74
+ "iter": 4000,
75
+ "eval_sec": 6.8,
76
+ "n_dets": 2380
77
+ },
78
+ {
79
+ "AP": 33.94447762278923,
80
+ "AP50": 58.09631015404435,
81
+ "AP75": 35.61235786807039,
82
+ "AP_cat1": 33.94447762278923,
83
+ "iter": 4500,
84
+ "eval_sec": 5.7,
85
+ "n_dets": 2340
86
+ },
87
+ {
88
+ "AP": 34.153097459756154,
89
+ "AP50": 58.53941212805452,
90
+ "AP75": 35.62155219060164,
91
+ "AP_cat1": 34.153097459756154,
92
+ "iter": 5000,
93
+ "eval_sec": 7.3,
94
+ "n_dets": 2322
95
+ },
96
+ {
97
+ "AP": 34.098399236266545,
98
+ "AP50": 58.47722636401763,
99
+ "AP75": 35.69835447716946,
100
+ "AP_cat1": 34.098399236266545,
101
+ "iter": 5500,
102
+ "eval_sec": 8.3,
103
+ "n_dets": 2302
104
+ },
105
+ {
106
+ "AP": 34.10685778918478,
107
+ "AP50": 58.455800860886356,
108
+ "AP75": 35.66927086771584,
109
+ "AP_cat1": 34.10685778918478,
110
+ "iter": 6000,
111
+ "eval_sec": 7.9,
112
+ "n_dets": 2299
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed3/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__mask_B10288_s3
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_full/budget_mask_B10288/seed3
273
+ SEED: 3
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_full/budget_mask_B10288/seed3/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_full",
4
+ "setting": "budget_mask_B10288",
5
+ "seed": 3,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/budget/mask_B10288_s3.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 4289.8,
12
+ "best_iter": 500,
13
+ "best_val": 53.59327078108764,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 3663,
18
+ "sec": 8.2
19
+ },
20
+ "test": {
21
+ "n_dets": 3689,
22
+ "sec": 9.9
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed3/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 68.14751031756715,
9
+ "F1@0.5": 82.20777055194264
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 68.49634917659466,
14
+ "F1@0.5": 82.61161463578382
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 68.79631270663168,
19
+ "F1@0.5": 82.94495103005741
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 69.26805688326493,
24
+ "F1@0.5": 83.50076621828708
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 69.39998818818661,
29
+ "F1@0.5": 83.6326321190354
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 69.54121847855788,
34
+ "F1@0.5": 83.76801647220316
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 69.7949764678759,
39
+ "F1@0.5": 84.06147470212399
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 70.09275150403418,
44
+ "F1@0.5": 84.2949276625414
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 70.29922087413034,
49
+ "F1@0.5": 84.49666783584708
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed3/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 53.59327078108764,
4
+ "best_iter": 500,
5
+ "history": [
6
+ {
7
+ "AP": 53.59327078108764,
8
+ "AP50": 83.39350323515929,
9
+ "AP75": 58.00673412777327,
10
+ "AP_cat1": 53.59327078108764,
11
+ "iter": 500,
12
+ "eval_sec": 11.0,
13
+ "n_dets": 3663
14
+ },
15
+ {
16
+ "AP": 51.43137036116261,
17
+ "AP50": 82.08787573271665,
18
+ "AP75": 55.13455973401984,
19
+ "AP_cat1": 51.43137036116261,
20
+ "iter": 1000,
21
+ "eval_sec": 9.3,
22
+ "n_dets": 3096
23
+ },
24
+ {
25
+ "AP": 50.66312711078466,
26
+ "AP50": 83.0466219439879,
27
+ "AP75": 54.17966769550383,
28
+ "AP_cat1": 50.66312711078466,
29
+ "iter": 1500,
30
+ "eval_sec": 9.6,
31
+ "n_dets": 3117
32
+ },
33
+ {
34
+ "AP": 52.80069893212615,
35
+ "AP50": 83.13761182560383,
36
+ "AP75": 57.45773112107682,
37
+ "AP_cat1": 52.80069893212615,
38
+ "iter": 2000,
39
+ "eval_sec": 9.4,
40
+ "n_dets": 3012
41
+ },
42
+ {
43
+ "AP": 53.17191538868179,
44
+ "AP50": 82.40857499041661,
45
+ "AP75": 57.9471470382699,
46
+ "AP_cat1": 53.17191538868179,
47
+ "iter": 2500,
48
+ "eval_sec": 7.9,
49
+ "n_dets": 3036
50
+ },
51
+ {
52
+ "AP": 51.951994511438706,
53
+ "AP50": 82.28994333272506,
54
+ "AP75": 55.717034843988685,
55
+ "AP_cat1": 51.951994511438706,
56
+ "iter": 3000,
57
+ "eval_sec": 7.3,
58
+ "n_dets": 2959
59
+ },
60
+ {
61
+ "AP": 53.189764052750085,
62
+ "AP50": 82.44730866260764,
63
+ "AP75": 57.86297566604247,
64
+ "AP_cat1": 53.189764052750085,
65
+ "iter": 3500,
66
+ "eval_sec": 6.5,
67
+ "n_dets": 2954
68
+ },
69
+ {
70
+ "AP": 53.50661107635657,
71
+ "AP50": 82.54903363991737,
72
+ "AP75": 57.973447095018905,
73
+ "AP_cat1": 53.50661107635657,
74
+ "iter": 4000,
75
+ "eval_sec": 9.2,
76
+ "n_dets": 2970
77
+ },
78
+ {
79
+ "AP": 53.57509273850696,
80
+ "AP50": 82.23988719256702,
81
+ "AP75": 58.017002966411745,
82
+ "AP_cat1": 53.57509273850696,
83
+ "iter": 4500,
84
+ "eval_sec": 4.5,
85
+ "n_dets": 2950
86
+ },
87
+ {
88
+ "AP": 53.054520272099595,
89
+ "AP50": 82.25893391323186,
90
+ "AP75": 57.78139123328647,
91
+ "AP_cat1": 53.054520272099595,
92
+ "iter": 5000,
93
+ "eval_sec": 4.1,
94
+ "n_dets": 2915
95
+ },
96
+ {
97
+ "AP": 53.09570105511043,
98
+ "AP50": 82.2723674306724,
99
+ "AP75": 57.90991252650027,
100
+ "AP_cat1": 53.09570105511043,
101
+ "iter": 5500,
102
+ "eval_sec": 4.4,
103
+ "n_dets": 2903
104
+ },
105
+ {
106
+ "AP": 53.074237529981936,
107
+ "AP50": 82.3038732738104,
108
+ "AP75": 57.869232345825274,
109
+ "AP_cat1": 53.074237529981936,
110
+ "iter": 6000,
111
+ "eval_sec": 8.3,
112
+ "n_dets": 2904
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed4/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__mask_B10288_s4
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_full/budget_mask_B10288/seed4
273
+ SEED: 4
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_full/budget_mask_B10288/seed4/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_full",
4
+ "setting": "budget_mask_B10288",
5
+ "seed": 4,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/budget/mask_B10288_s4.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 6809.8,
12
+ "best_iter": 1000,
13
+ "best_val": 55.683939244727796,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 3565,
18
+ "sec": 12.6
19
+ },
20
+ "test": {
21
+ "n_dets": 3071,
22
+ "sec": 13.2
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed4/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 68.75968736157915,
9
+ "F1@0.5": 82.78580814717476
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 68.99037047662935,
14
+ "F1@0.5": 83.02509907529722
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 69.27189591839215,
19
+ "F1@0.5": 83.33333333333334
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 69.57124157547874,
24
+ "F1@0.5": 83.65882156568185
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 69.75720035377152,
29
+ "F1@0.5": 83.81784574021172
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 69.93183472464283,
34
+ "F1@0.5": 83.94727948631294
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 70.12082846119424,
39
+ "F1@0.5": 84.12104726283577
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 70.32303460928343,
44
+ "F1@0.5": 84.28082191780823
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 70.59870472821042,
49
+ "F1@0.5": 84.55704929334712
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_full/budget_mask_B10288/seed4/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 55.683939244727796,
4
+ "best_iter": 1000,
5
+ "history": [
6
+ {
7
+ "AP": 55.53730574350054,
8
+ "AP50": 84.49639410680062,
9
+ "AP75": 61.788841668024084,
10
+ "AP_cat1": 55.53730574350054,
11
+ "iter": 500,
12
+ "eval_sec": 13.3,
13
+ "n_dets": 4296
14
+ },
15
+ {
16
+ "AP": 55.683939244727796,
17
+ "AP50": 84.14730388137609,
18
+ "AP75": 62.595574032731434,
19
+ "AP_cat1": 55.683939244727796,
20
+ "iter": 1000,
21
+ "eval_sec": 12.1,
22
+ "n_dets": 3565
23
+ },
24
+ {
25
+ "AP": 55.655582475148535,
26
+ "AP50": 84.40798575816213,
27
+ "AP75": 62.41053572874732,
28
+ "AP_cat1": 55.655582475148535,
29
+ "iter": 1500,
30
+ "eval_sec": 11.5,
31
+ "n_dets": 3365
32
+ },
33
+ {
34
+ "AP": 54.47318368276821,
35
+ "AP50": 84.70947524517887,
36
+ "AP75": 60.374700861342156,
37
+ "AP_cat1": 54.47318368276821,
38
+ "iter": 2000,
39
+ "eval_sec": 11.5,
40
+ "n_dets": 3319
41
+ },
42
+ {
43
+ "AP": 54.534234389651026,
44
+ "AP50": 84.15033078005109,
45
+ "AP75": 60.555963973166286,
46
+ "AP_cat1": 54.534234389651026,
47
+ "iter": 2500,
48
+ "eval_sec": 11.9,
49
+ "n_dets": 3355
50
+ },
51
+ {
52
+ "AP": 55.05516416098729,
53
+ "AP50": 84.28362625048061,
54
+ "AP75": 61.47250603109418,
55
+ "AP_cat1": 55.05516416098729,
56
+ "iter": 3000,
57
+ "eval_sec": 10.6,
58
+ "n_dets": 3300
59
+ },
60
+ {
61
+ "AP": 54.29870253796304,
62
+ "AP50": 83.32069845421029,
63
+ "AP75": 60.336435518637984,
64
+ "AP_cat1": 54.29870253796304,
65
+ "iter": 3500,
66
+ "eval_sec": 14.4,
67
+ "n_dets": 3248
68
+ },
69
+ {
70
+ "AP": 54.23191863381992,
71
+ "AP50": 83.989146879194,
72
+ "AP75": 60.6694855471349,
73
+ "AP_cat1": 54.23191863381992,
74
+ "iter": 4000,
75
+ "eval_sec": 14.2,
76
+ "n_dets": 3271
77
+ },
78
+ {
79
+ "AP": 54.19909834959963,
80
+ "AP50": 83.92259195525399,
81
+ "AP75": 60.595534173493384,
82
+ "AP_cat1": 54.19909834959963,
83
+ "iter": 4500,
84
+ "eval_sec": 13.9,
85
+ "n_dets": 3300
86
+ },
87
+ {
88
+ "AP": 53.51562482587822,
89
+ "AP50": 82.99289417102895,
90
+ "AP75": 59.97363566608247,
91
+ "AP_cat1": 53.51562482587822,
92
+ "iter": 5000,
93
+ "eval_sec": 13.7,
94
+ "n_dets": 3230
95
+ },
96
+ {
97
+ "AP": 53.35978320691517,
98
+ "AP50": 83.16132007771043,
99
+ "AP75": 60.01734938908766,
100
+ "AP_cat1": 53.35978320691517,
101
+ "iter": 5500,
102
+ "eval_sec": 13.0,
103
+ "n_dets": 3221
104
+ },
105
+ {
106
+ "AP": 53.33185137837211,
107
+ "AP50": 82.97407269646526,
108
+ "AP75": 59.92873233324574,
109
+ "AP_cat1": 53.33185137837211,
110
+ "iter": 6000,
111
+ "eval_sec": 12.4,
112
+ "n_dets": 3214
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed0/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__mask_B164605_s0
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_full/budget_mask_B164605/seed0
273
+ SEED: 0
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_full/budget_mask_B164605/seed0/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_full",
4
+ "setting": "budget_mask_B164605",
5
+ "seed": 0,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/budget/mask_B164605_s0.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 3227.1,
12
+ "best_iter": 1000,
13
+ "best_val": 61.715257237539525,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 4850,
18
+ "sec": 5.2
19
+ },
20
+ "test": {
21
+ "n_dets": 5488,
22
+ "sec": 10.2
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed0/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 71.86990511976931,
9
+ "F1@0.5": 84.81540930979133
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 72.52297597275773,
14
+ "F1@0.5": 85.54725560246833
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 73.00230046805979,
19
+ "F1@0.5": 86.0751189109398
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 73.37894634963834,
24
+ "F1@0.5": 86.46951924665454
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 73.9583322666201,
29
+ "F1@0.5": 87.09893048128342
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 74.32371201995484,
34
+ "F1@0.5": 87.45991561181434
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 74.68538005670767,
39
+ "F1@0.5": 87.78573865574889
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 74.80840366375688,
44
+ "F1@0.5": 87.8124461299776
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 74.93851708303966,
49
+ "F1@0.5": 87.80146801817547
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed0/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 61.715257237539525,
4
+ "best_iter": 1000,
5
+ "history": [
6
+ {
7
+ "AP": 59.77339828032348,
8
+ "AP50": 89.30276934032464,
9
+ "AP75": 66.000540582682,
10
+ "AP_cat1": 59.77339828032348,
11
+ "iter": 500,
12
+ "eval_sec": 5.6,
13
+ "n_dets": 5433
14
+ },
15
+ {
16
+ "AP": 61.715257237539525,
17
+ "AP50": 89.80432359367444,
18
+ "AP75": 69.3276088710589,
19
+ "AP_cat1": 61.715257237539525,
20
+ "iter": 1000,
21
+ "eval_sec": 5.2,
22
+ "n_dets": 4850
23
+ },
24
+ {
25
+ "AP": 60.596718062723674,
26
+ "AP50": 88.93815034224521,
27
+ "AP75": 68.69696813547843,
28
+ "AP_cat1": 60.596718062723674,
29
+ "iter": 1500,
30
+ "eval_sec": 4.9,
31
+ "n_dets": 3890
32
+ },
33
+ {
34
+ "AP": 60.066282930378456,
35
+ "AP50": 88.12963062700011,
36
+ "AP75": 66.19392919135169,
37
+ "AP_cat1": 60.066282930378456,
38
+ "iter": 2000,
39
+ "eval_sec": 4.9,
40
+ "n_dets": 3819
41
+ },
42
+ {
43
+ "AP": 60.34525687082927,
44
+ "AP50": 88.37591298003828,
45
+ "AP75": 67.05931842165593,
46
+ "AP_cat1": 60.34525687082927,
47
+ "iter": 2500,
48
+ "eval_sec": 4.9,
49
+ "n_dets": 3706
50
+ },
51
+ {
52
+ "AP": 60.7659130051574,
53
+ "AP50": 87.53363421085523,
54
+ "AP75": 67.53633639261038,
55
+ "AP_cat1": 60.7659130051574,
56
+ "iter": 3000,
57
+ "eval_sec": 4.7,
58
+ "n_dets": 3243
59
+ },
60
+ {
61
+ "AP": 59.58620598750636,
62
+ "AP50": 87.18659383166812,
63
+ "AP75": 66.66969035227872,
64
+ "AP_cat1": 59.58620598750636,
65
+ "iter": 3500,
66
+ "eval_sec": 4.9,
67
+ "n_dets": 3311
68
+ },
69
+ {
70
+ "AP": 59.757011756852606,
71
+ "AP50": 87.44851895163039,
72
+ "AP75": 67.33377737244777,
73
+ "AP_cat1": 59.757011756852606,
74
+ "iter": 4000,
75
+ "eval_sec": 4.8,
76
+ "n_dets": 3339
77
+ },
78
+ {
79
+ "AP": 58.035962640498894,
80
+ "AP50": 87.7371492966872,
81
+ "AP75": 64.39084908154351,
82
+ "AP_cat1": 58.035962640498894,
83
+ "iter": 4500,
84
+ "eval_sec": 4.9,
85
+ "n_dets": 3446
86
+ },
87
+ {
88
+ "AP": 58.93170194792097,
89
+ "AP50": 87.30879734624081,
90
+ "AP75": 66.15968619422996,
91
+ "AP_cat1": 58.93170194792097,
92
+ "iter": 5000,
93
+ "eval_sec": 4.8,
94
+ "n_dets": 3210
95
+ },
96
+ {
97
+ "AP": 58.84210587610483,
98
+ "AP50": 87.0203704507666,
99
+ "AP75": 65.78749226417895,
100
+ "AP_cat1": 58.84210587610483,
101
+ "iter": 5500,
102
+ "eval_sec": 4.7,
103
+ "n_dets": 3166
104
+ },
105
+ {
106
+ "AP": 58.59381757302199,
107
+ "AP50": 86.9754329442704,
108
+ "AP75": 65.69025606408482,
109
+ "AP_cat1": 58.59381757302199,
110
+ "iter": 6000,
111
+ "eval_sec": 4.8,
112
+ "n_dets": 3164
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed1/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__mask_B164605_s1
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_full/budget_mask_B164605/seed1
273
+ SEED: 1
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_full/budget_mask_B164605/seed1/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_full",
4
+ "setting": "budget_mask_B164605",
5
+ "seed": 1,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/budget/mask_B164605_s1.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 5972.5,
12
+ "best_iter": 3000,
13
+ "best_val": 61.04560979087882,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 3427,
18
+ "sec": 4.4
19
+ },
20
+ "test": {
21
+ "n_dets": 3466,
22
+ "sec": 6.4
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed1/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 74.5857926443151,
9
+ "F1@0.5": 87.3893435777518
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 74.88588544380161,
14
+ "F1@0.5": 87.71222054126744
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 75.08552950028663,
19
+ "F1@0.5": 87.92841465473578
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 75.38378956041336,
24
+ "F1@0.5": 88.26625912718627
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 75.61248928971375,
29
+ "F1@0.5": 88.50613915416098
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 75.73420516928606,
34
+ "F1@0.5": 88.5954381752701
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 75.98519009250032,
39
+ "F1@0.5": 88.8505350362444
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 76.0622949213048,
44
+ "F1@0.5": 88.9196675900277
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 76.15046446104436,
49
+ "F1@0.5": 88.96623738252698
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed1/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 61.04560979087882,
4
+ "best_iter": 3000,
5
+ "history": [
6
+ {
7
+ "AP": 59.346953782989054,
8
+ "AP50": 88.91920375938066,
9
+ "AP75": 67.08680546600162,
10
+ "AP_cat1": 59.346953782989054,
11
+ "iter": 500,
12
+ "eval_sec": 15.5,
13
+ "n_dets": 5561
14
+ },
15
+ {
16
+ "AP": 59.9111292032113,
17
+ "AP50": 90.42507029822767,
18
+ "AP75": 69.56506452086164,
19
+ "AP_cat1": 59.9111292032113,
20
+ "iter": 1000,
21
+ "eval_sec": 14.8,
22
+ "n_dets": 4610
23
+ },
24
+ {
25
+ "AP": 60.257102055245745,
26
+ "AP50": 90.31348393727036,
27
+ "AP75": 68.9219543833754,
28
+ "AP_cat1": 60.257102055245745,
29
+ "iter": 1500,
30
+ "eval_sec": 14.7,
31
+ "n_dets": 4710
32
+ },
33
+ {
34
+ "AP": 60.980104117678245,
35
+ "AP50": 88.28800233375728,
36
+ "AP75": 68.39273459766706,
37
+ "AP_cat1": 60.980104117678245,
38
+ "iter": 2000,
39
+ "eval_sec": 14.3,
40
+ "n_dets": 3482
41
+ },
42
+ {
43
+ "AP": 59.60121960834138,
44
+ "AP50": 88.67075901039432,
45
+ "AP75": 66.87804721055102,
46
+ "AP_cat1": 59.60121960834138,
47
+ "iter": 2500,
48
+ "eval_sec": 7.4,
49
+ "n_dets": 3675
50
+ },
51
+ {
52
+ "AP": 61.04560979087882,
53
+ "AP50": 89.37118848789886,
54
+ "AP75": 68.29513734237067,
55
+ "AP_cat1": 61.04560979087882,
56
+ "iter": 3000,
57
+ "eval_sec": 4.7,
58
+ "n_dets": 3427
59
+ },
60
+ {
61
+ "AP": 58.83042340980828,
62
+ "AP50": 87.73027951675327,
63
+ "AP75": 66.88811787914403,
64
+ "AP_cat1": 58.83042340980828,
65
+ "iter": 3500,
66
+ "eval_sec": 4.7,
67
+ "n_dets": 3560
68
+ },
69
+ {
70
+ "AP": 59.74085458924168,
71
+ "AP50": 88.18992600048512,
72
+ "AP75": 67.50882000498191,
73
+ "AP_cat1": 59.74085458924168,
74
+ "iter": 4000,
75
+ "eval_sec": 4.5,
76
+ "n_dets": 3438
77
+ },
78
+ {
79
+ "AP": 59.84455169221021,
80
+ "AP50": 88.01938072930207,
81
+ "AP75": 67.81891343389704,
82
+ "AP_cat1": 59.84455169221021,
83
+ "iter": 4500,
84
+ "eval_sec": 4.6,
85
+ "n_dets": 3321
86
+ },
87
+ {
88
+ "AP": 58.870636857772936,
89
+ "AP50": 87.81248500814756,
90
+ "AP75": 66.59529076554558,
91
+ "AP_cat1": 58.870636857772936,
92
+ "iter": 5000,
93
+ "eval_sec": 4.5,
94
+ "n_dets": 3268
95
+ },
96
+ {
97
+ "AP": 58.96496951871421,
98
+ "AP50": 87.85417852876766,
99
+ "AP75": 66.58697652372842,
100
+ "AP_cat1": 58.96496951871421,
101
+ "iter": 5500,
102
+ "eval_sec": 4.5,
103
+ "n_dets": 3240
104
+ },
105
+ {
106
+ "AP": 59.00630386095821,
107
+ "AP50": 87.77548237730278,
108
+ "AP75": 66.46554332277047,
109
+ "AP_cat1": 59.00630386095821,
110
+ "iter": 6000,
111
+ "eval_sec": 4.5,
112
+ "n_dets": 3241
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed2/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__mask_B164605_s2
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_full/budget_mask_B164605/seed2
273
+ SEED: 2
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_full/budget_mask_B164605/seed2/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_full",
4
+ "setting": "budget_mask_B164605",
5
+ "seed": 2,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/budget/mask_B164605_s2.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 2928.8,
12
+ "best_iter": 1000,
13
+ "best_val": 62.766535963464165,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 5186,
18
+ "sec": 5.1
19
+ },
20
+ "test": {
21
+ "n_dets": 5606,
22
+ "sec": 10.0
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed2/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 71.31355730608605,
9
+ "F1@0.5": 83.8543345149855
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 71.91108298388654,
14
+ "F1@0.5": 84.54782042940793
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 72.56939536939475,
19
+ "F1@0.5": 85.2965335961886
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 73.13151737760472,
24
+ "F1@0.5": 85.91315876698708
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 73.56499854214572,
29
+ "F1@0.5": 86.3871722064473
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 74.05961200764992,
34
+ "F1@0.5": 86.88469318948077
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 74.39592180751342,
39
+ "F1@0.5": 87.23259762308999
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 74.69338103908059,
44
+ "F1@0.5": 87.52353243196987
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 74.83869882291502,
49
+ "F1@0.5": 87.65474552957359
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed2/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 62.766535963464165,
4
+ "best_iter": 1000,
5
+ "history": [
6
+ {
7
+ "AP": 61.21574769754267,
8
+ "AP50": 89.32139656006687,
9
+ "AP75": 68.42434416152888,
10
+ "AP_cat1": 61.21574769754267,
11
+ "iter": 500,
12
+ "eval_sec": 4.7,
13
+ "n_dets": 5294
14
+ },
15
+ {
16
+ "AP": 62.766535963464165,
17
+ "AP50": 89.84324327379653,
18
+ "AP75": 70.97852608762267,
19
+ "AP_cat1": 62.766535963464165,
20
+ "iter": 1000,
21
+ "eval_sec": 4.9,
22
+ "n_dets": 5186
23
+ },
24
+ {
25
+ "AP": 60.19194194113049,
26
+ "AP50": 90.29450442868243,
27
+ "AP75": 66.77910159734289,
28
+ "AP_cat1": 60.19194194113049,
29
+ "iter": 1500,
30
+ "eval_sec": 4.5,
31
+ "n_dets": 4239
32
+ },
33
+ {
34
+ "AP": 60.72273419846132,
35
+ "AP50": 89.83892795855056,
36
+ "AP75": 68.98717794706938,
37
+ "AP_cat1": 60.72273419846132,
38
+ "iter": 2000,
39
+ "eval_sec": 4.7,
40
+ "n_dets": 4322
41
+ },
42
+ {
43
+ "AP": 59.44227312066525,
44
+ "AP50": 89.07126316835216,
45
+ "AP75": 67.53558746721696,
46
+ "AP_cat1": 59.44227312066525,
47
+ "iter": 2500,
48
+ "eval_sec": 4.4,
49
+ "n_dets": 3792
50
+ },
51
+ {
52
+ "AP": 59.035890940010006,
53
+ "AP50": 88.3582412744769,
54
+ "AP75": 66.20780506695222,
55
+ "AP_cat1": 59.035890940010006,
56
+ "iter": 3000,
57
+ "eval_sec": 4.5,
58
+ "n_dets": 4180
59
+ },
60
+ {
61
+ "AP": 59.708227904415864,
62
+ "AP50": 88.83949078147198,
63
+ "AP75": 67.56401899022487,
64
+ "AP_cat1": 59.708227904415864,
65
+ "iter": 3500,
66
+ "eval_sec": 4.2,
67
+ "n_dets": 3563
68
+ },
69
+ {
70
+ "AP": 58.98138466692413,
71
+ "AP50": 88.68336321022025,
72
+ "AP75": 66.74963686010196,
73
+ "AP_cat1": 58.98138466692413,
74
+ "iter": 4000,
75
+ "eval_sec": 5.2,
76
+ "n_dets": 3459
77
+ },
78
+ {
79
+ "AP": 55.732028761259436,
80
+ "AP50": 86.82257700623907,
81
+ "AP75": 61.69419338037792,
82
+ "AP_cat1": 55.732028761259436,
83
+ "iter": 4500,
84
+ "eval_sec": 4.7,
85
+ "n_dets": 3442
86
+ },
87
+ {
88
+ "AP": 57.61639546273296,
89
+ "AP50": 87.30117802546015,
90
+ "AP75": 65.01170091600767,
91
+ "AP_cat1": 57.61639546273296,
92
+ "iter": 5000,
93
+ "eval_sec": 4.6,
94
+ "n_dets": 3341
95
+ },
96
+ {
97
+ "AP": 57.861090130716555,
98
+ "AP50": 87.54933852582245,
99
+ "AP75": 65.35737688676797,
100
+ "AP_cat1": 57.861090130716555,
101
+ "iter": 5500,
102
+ "eval_sec": 4.8,
103
+ "n_dets": 3279
104
+ },
105
+ {
106
+ "AP": 57.844629965664716,
107
+ "AP50": 87.4619993158885,
108
+ "AP75": 65.39585835294055,
109
+ "AP_cat1": 57.844629965664716,
110
+ "iter": 6000,
111
+ "eval_sec": 4.7,
112
+ "n_dets": 3280
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed3/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__mask_B164605_s3
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_full/budget_mask_B164605/seed3
273
+ SEED: 3
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_full/budget_mask_B164605/seed3/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_full",
4
+ "setting": "budget_mask_B164605",
5
+ "seed": 3,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/budget/mask_B164605_s3.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 2997.5,
12
+ "best_iter": 500,
13
+ "best_val": 62.01388332425404,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 5512,
18
+ "sec": 4.8
19
+ },
20
+ "test": {
21
+ "n_dets": 5835,
22
+ "sec": 9.0
23
+ }
24
+ }
25
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed3/score_thr.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "score_thr": 0.7,
3
+ "chosen_on": "val",
4
+ "criterion": "PQ",
5
+ "sweep": [
6
+ {
7
+ "thr": 0.3,
8
+ "PQ": 70.0389212269996,
9
+ "F1@0.5": 83.12927333439339
10
+ },
11
+ {
12
+ "thr": 0.35,
13
+ "PQ": 70.92461612630416,
14
+ "F1@0.5": 84.17150225660865
15
+ },
16
+ {
17
+ "thr": 0.4,
18
+ "PQ": 71.71651105556018,
19
+ "F1@0.5": 85.06535947712418
20
+ },
21
+ {
22
+ "thr": 0.45,
23
+ "PQ": 72.33231304109556,
24
+ "F1@0.5": 85.7331790378575
25
+ },
26
+ {
27
+ "thr": 0.5,
28
+ "PQ": 73.03691260264594,
29
+ "F1@0.5": 86.52173913043478
30
+ },
31
+ {
32
+ "thr": 0.55,
33
+ "PQ": 73.34086603184453,
34
+ "F1@0.5": 86.80836707152496
35
+ },
36
+ {
37
+ "thr": 0.6,
38
+ "PQ": 73.84188949341132,
39
+ "F1@0.5": 87.31865506059054
40
+ },
41
+ {
42
+ "thr": 0.65,
43
+ "PQ": 73.95749081029945,
44
+ "F1@0.5": 87.27774900742276
45
+ },
46
+ {
47
+ "thr": 0.7,
48
+ "PQ": 74.13809003165615,
49
+ "F1@0.5": 87.37999650898935
50
+ }
51
+ ]
52
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed3/val_history.json ADDED
@@ -0,0 +1,115 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metric": "AP",
3
+ "best": 62.01388332425404,
4
+ "best_iter": 500,
5
+ "history": [
6
+ {
7
+ "AP": 62.01388332425404,
8
+ "AP50": 90.49320364952965,
9
+ "AP75": 69.2398523986648,
10
+ "AP_cat1": 62.01388332425404,
11
+ "iter": 500,
12
+ "eval_sec": 4.7,
13
+ "n_dets": 5512
14
+ },
15
+ {
16
+ "AP": 60.25533711231912,
17
+ "AP50": 90.30678433069058,
18
+ "AP75": 68.01751194564594,
19
+ "AP_cat1": 60.25533711231912,
20
+ "iter": 1000,
21
+ "eval_sec": 4.6,
22
+ "n_dets": 4825
23
+ },
24
+ {
25
+ "AP": 59.58736127754247,
26
+ "AP50": 90.0317787985166,
27
+ "AP75": 66.84559327282093,
28
+ "AP_cat1": 59.58736127754247,
29
+ "iter": 1500,
30
+ "eval_sec": 4.6,
31
+ "n_dets": 4816
32
+ },
33
+ {
34
+ "AP": 58.791179593961054,
35
+ "AP50": 89.97169671714171,
36
+ "AP75": 66.84081426836642,
37
+ "AP_cat1": 58.791179593961054,
38
+ "iter": 2000,
39
+ "eval_sec": 4.4,
40
+ "n_dets": 4428
41
+ },
42
+ {
43
+ "AP": 60.108649453205985,
44
+ "AP50": 89.06893121174426,
45
+ "AP75": 67.25399709889392,
46
+ "AP_cat1": 60.108649453205985,
47
+ "iter": 2500,
48
+ "eval_sec": 4.5,
49
+ "n_dets": 4308
50
+ },
51
+ {
52
+ "AP": 57.04538820512215,
53
+ "AP50": 88.62741605958831,
54
+ "AP75": 63.093111760234486,
55
+ "AP_cat1": 57.04538820512215,
56
+ "iter": 3000,
57
+ "eval_sec": 4.4,
58
+ "n_dets": 3855
59
+ },
60
+ {
61
+ "AP": 59.56778429454046,
62
+ "AP50": 88.53993909746356,
63
+ "AP75": 67.56840792623947,
64
+ "AP_cat1": 59.56778429454046,
65
+ "iter": 3500,
66
+ "eval_sec": 4.5,
67
+ "n_dets": 3784
68
+ },
69
+ {
70
+ "AP": 58.510361980092874,
71
+ "AP50": 87.83595830478033,
72
+ "AP75": 66.11988870227492,
73
+ "AP_cat1": 58.510361980092874,
74
+ "iter": 4000,
75
+ "eval_sec": 4.2,
76
+ "n_dets": 3446
77
+ },
78
+ {
79
+ "AP": 58.348698849987166,
80
+ "AP50": 88.41588238684585,
81
+ "AP75": 66.72056854125232,
82
+ "AP_cat1": 58.348698849987166,
83
+ "iter": 4500,
84
+ "eval_sec": 4.5,
85
+ "n_dets": 3523
86
+ },
87
+ {
88
+ "AP": 59.22155173117305,
89
+ "AP50": 88.00507464920143,
90
+ "AP75": 67.15272462708818,
91
+ "AP_cat1": 59.22155173117305,
92
+ "iter": 5000,
93
+ "eval_sec": 4.2,
94
+ "n_dets": 3367
95
+ },
96
+ {
97
+ "AP": 59.069852147700196,
98
+ "AP50": 88.1132760267318,
99
+ "AP75": 66.56764932316398,
100
+ "AP_cat1": 59.069852147700196,
101
+ "iter": 5500,
102
+ "eval_sec": 4.3,
103
+ "n_dets": 3335
104
+ },
105
+ {
106
+ "AP": 58.950297923274555,
107
+ "AP50": 88.12787432275867,
108
+ "AP75": 66.4679812015891,
109
+ "AP_cat1": 58.950297923274555,
110
+ "iter": 6000,
111
+ "eval_sec": 4.4,
112
+ "n_dets": 3346
113
+ }
114
+ ]
115
+ }
dsb2018/mrcnn_full/budget_mask_B164605/seed4/config.yaml ADDED
@@ -0,0 +1,324 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 6
6
+ REPEAT_THRESHOLD: 0.0
7
+ SAMPLER_TRAIN: TrainingSampler
8
+ DATASETS:
9
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
10
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
11
+ PROPOSAL_FILES_TEST: []
12
+ PROPOSAL_FILES_TRAIN: []
13
+ TEST:
14
+ - dsb2018_val
15
+ TRAIN:
16
+ - dsb2018__mask_B164605_s4
17
+ GLOBAL:
18
+ HACK: 1.0
19
+ INPUT:
20
+ CROP:
21
+ ENABLED: false
22
+ SIZE:
23
+ - 0.9
24
+ - 0.9
25
+ TYPE: relative_range
26
+ FORMAT: BGR
27
+ MASK_FORMAT: bitmask
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 640
31
+ MIN_SIZE_TRAIN:
32
+ - 512
33
+ - 576
34
+ - 640
35
+ - 704
36
+ - 768
37
+ MIN_SIZE_TRAIN_SAMPLING: choice
38
+ POINT_SUP: false
39
+ RANDOM_FLIP: none
40
+ SAMPLE_POINTS: 0
41
+ MODEL:
42
+ ANCHOR_GENERATOR:
43
+ ANGLES:
44
+ - - -90
45
+ - 0
46
+ - 90
47
+ ASPECT_RATIOS:
48
+ - - 0.5
49
+ - 1.0
50
+ - 2.0
51
+ NAME: DefaultAnchorGenerator
52
+ OFFSET: 0.0
53
+ SIZES:
54
+ - - 8
55
+ - - 16
56
+ - - 32
57
+ - - 64
58
+ - - 128
59
+ BACKBONE:
60
+ FREEZE_AT: 2
61
+ NAME: build_resnet_fpn_backbone
62
+ DEVICE: cuda
63
+ FPN:
64
+ FUSE_TYPE: sum
65
+ IN_FEATURES:
66
+ - res2
67
+ - res3
68
+ - res4
69
+ - res5
70
+ NORM: ''
71
+ OUT_CHANNELS: 256
72
+ KEYPOINT_ON: false
73
+ LOAD_PROPOSALS: false
74
+ MASK_ON: true
75
+ META_ARCHITECTURE: GeneralizedRCNN
76
+ PANOPTIC_FPN:
77
+ COMBINE:
78
+ ENABLED: true
79
+ INSTANCES_CONFIDENCE_THRESH: 0.5
80
+ OVERLAP_THRESH: 0.5
81
+ STUFF_AREA_LIMIT: 4096
82
+ INSTANCE_LOSS_WEIGHT: 1.0
83
+ PIXEL_MEAN:
84
+ - 103.53
85
+ - 116.28
86
+ - 123.675
87
+ PIXEL_STD:
88
+ - 1.0
89
+ - 1.0
90
+ - 1.0
91
+ PROPOSAL_GENERATOR:
92
+ MIN_SIZE: 0
93
+ NAME: RPN
94
+ RESNETS:
95
+ AVD: false
96
+ AVG_DOWN: false
97
+ BOTTLENECK_WIDTH: 64
98
+ DEEP_STEM: false
99
+ DEFORM_MODULATED: false
100
+ DEFORM_NUM_GROUPS: 1
101
+ DEFORM_ON_PER_STAGE:
102
+ - false
103
+ - false
104
+ - false
105
+ - false
106
+ DEPTH: 50
107
+ NORM: FrozenBN
108
+ NUM_GROUPS: 1
109
+ OUT_FEATURES:
110
+ - res2
111
+ - res3
112
+ - res4
113
+ - res5
114
+ RADIX: 1
115
+ RES2_OUT_CHANNELS: 256
116
+ RES5_DILATION: 1
117
+ STEM_OUT_CHANNELS: 64
118
+ STRIDE_IN_1X1: true
119
+ WIDTH_PER_GROUP: 64
120
+ RETINANET:
121
+ BBOX_REG_WEIGHTS: &id002
122
+ - 1.0
123
+ - 1.0
124
+ - 1.0
125
+ - 1.0
126
+ FOCAL_LOSS_ALPHA: 0.25
127
+ FOCAL_LOSS_GAMMA: 2.0
128
+ IN_FEATURES:
129
+ - p3
130
+ - p4
131
+ - p5
132
+ - p6
133
+ - p7
134
+ IOU_LABELS:
135
+ - 0
136
+ - -1
137
+ - 1
138
+ IOU_THRESHOLDS:
139
+ - 0.4
140
+ - 0.5
141
+ NMS_THRESH_TEST: 0.5
142
+ NUM_CLASSES: 80
143
+ NUM_CONVS: 4
144
+ PRIOR_PROB: 0.01
145
+ SCORE_THRESH_TEST: 0.05
146
+ SMOOTH_L1_LOSS_BETA: 0.1
147
+ TOPK_CANDIDATES_TEST: 1000
148
+ ROI_BOX_CASCADE_HEAD:
149
+ BBOX_REG_WEIGHTS:
150
+ - &id001
151
+ - 10.0
152
+ - 10.0
153
+ - 5.0
154
+ - 5.0
155
+ - - 20.0
156
+ - 20.0
157
+ - 10.0
158
+ - 10.0
159
+ - - 30.0
160
+ - 30.0
161
+ - 15.0
162
+ - 15.0
163
+ IOUS:
164
+ - 0.5
165
+ - 0.6
166
+ - 0.7
167
+ ROI_BOX_HEAD:
168
+ BBOX_REG_LOSS_TYPE: smooth_l1
169
+ BBOX_REG_LOSS_WEIGHT: 1.0
170
+ BBOX_REG_WEIGHTS: *id001
171
+ CLS_AGNOSTIC_BBOX_REG: false
172
+ CONV_DIM: 256
173
+ FC_DIM: 1024
174
+ NAME: FastRCNNConvFCHead
175
+ NORM: ''
176
+ NUM_CONV: 0
177
+ NUM_FC: 2
178
+ POOLER_RESOLUTION: 7
179
+ POOLER_SAMPLING_RATIO: 0
180
+ POOLER_TYPE: ROIAlignV2
181
+ SMOOTH_L1_BETA: 0.0
182
+ TRAIN_ON_PRED_BOXES: false
183
+ ROI_HEADS:
184
+ BATCH_SIZE_PER_IMAGE: 512
185
+ IN_FEATURES:
186
+ - p2
187
+ - p3
188
+ - p4
189
+ - p5
190
+ IOU_LABELS:
191
+ - 0
192
+ - 1
193
+ IOU_THRESHOLDS:
194
+ - 0.5
195
+ NAME: StandardROIHeads
196
+ NMS_THRESH_TEST: 0.5
197
+ NUM_CLASSES: 1
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.05
201
+ ROI_KEYPOINT_HEAD:
202
+ CONV_DIMS:
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ LOSS_WEIGHT: 1.0
212
+ MIN_KEYPOINTS_PER_IMAGE: 1
213
+ NAME: KRCNNConvDeconvUpsampleHead
214
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
215
+ NUM_KEYPOINTS: 17
216
+ POOLER_RESOLUTION: 14
217
+ POOLER_SAMPLING_RATIO: 0
218
+ POOLER_TYPE: ROIAlignV2
219
+ ROI_MASK_HEAD:
220
+ CLS_AGNOSTIC_MASK: false
221
+ CONV_DIM: 256
222
+ NAME: MaskRCNNConvUpsampleHead
223
+ NORM: ''
224
+ NUM_CONV: 4
225
+ POOLER_RESOLUTION: 14
226
+ POOLER_SAMPLING_RATIO: 0
227
+ POOLER_TYPE: ROIAlignV2
228
+ RPN:
229
+ BATCH_SIZE_PER_IMAGE: 256
230
+ BBOX_REG_LOSS_TYPE: smooth_l1
231
+ BBOX_REG_LOSS_WEIGHT: 1.0
232
+ BBOX_REG_WEIGHTS: *id002
233
+ BOUNDARY_THRESH: -1
234
+ CONV_DIMS:
235
+ - -1
236
+ HEAD_NAME: StandardRPNHead
237
+ IN_FEATURES:
238
+ - p2
239
+ - p3
240
+ - p4
241
+ - p5
242
+ - p6
243
+ IOU_LABELS:
244
+ - 0
245
+ - -1
246
+ - 1
247
+ IOU_THRESHOLDS:
248
+ - 0.3
249
+ - 0.7
250
+ LOSS_WEIGHT: 1.0
251
+ NMS_THRESH: 0.7
252
+ POSITIVE_FRACTION: 0.5
253
+ POST_NMS_TOPK_TEST: 2000
254
+ POST_NMS_TOPK_TRAIN: 2000
255
+ PRE_NMS_TOPK_TEST: 3000
256
+ PRE_NMS_TOPK_TRAIN: 4000
257
+ SMOOTH_L1_BETA: 0.0
258
+ SEM_SEG_HEAD:
259
+ COMMON_STRIDE: 4
260
+ CONVS_DIM: 128
261
+ IGNORE_VALUE: 255
262
+ IN_FEATURES:
263
+ - p2
264
+ - p3
265
+ - p4
266
+ - p5
267
+ LOSS_WEIGHT: 1.0
268
+ NAME: SemSegFPNHead
269
+ NORM: GN
270
+ NUM_CLASSES: 54
271
+ WEIGHTS: envs/ckpts/R-50.pkl
272
+ OUTPUT_DIR: runs/dsb2018/mrcnn_full/budget_mask_B164605/seed4
273
+ SEED: 4
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: true
277
+ BASE_LR: 0.02
278
+ BIAS_LR_FACTOR: 1.0
279
+ CHECKPOINT_PERIOD: 1000000
280
+ CLIP_GRADIENTS:
281
+ CLIP_TYPE: norm
282
+ CLIP_VALUE: 10.0
283
+ ENABLED: true
284
+ NORM_TYPE: 2.0
285
+ GAMMA: 0.1
286
+ IMS_PER_BATCH: 8
287
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
288
+ MAX_ITER: 6000
289
+ MOMENTUM: 0.9
290
+ NESTEROV: false
291
+ REFERENCE_WORLD_SIZE: 0
292
+ STEPS:
293
+ - 4500
294
+ - 5400
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 200
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: 0.0001
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 4000
306
+ MIN_SIZES:
307
+ - 400
308
+ - 500
309
+ - 600
310
+ - 700
311
+ - 800
312
+ - 900
313
+ - 1000
314
+ - 1100
315
+ - 1200
316
+ DETECTIONS_PER_IMAGE: 500
317
+ EVAL_PERIOD: 0
318
+ EXPECTED_RESULTS: []
319
+ KEYPOINT_OKS_SIGMAS: []
320
+ PRECISE_BN:
321
+ ENABLED: false
322
+ NUM_ITER: 200
323
+ VERSION: 2
324
+ VIS_PERIOD: 0
dsb2018/mrcnn_full/budget_mask_B164605/seed4/run_info.json ADDED
@@ -0,0 +1,25 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "dataset": "dsb2018",
3
+ "method": "mrcnn_full",
4
+ "setting": "budget_mask_B164605",
5
+ "seed": 4,
6
+ "profile": null,
7
+ "no_pl_weight": false,
8
+ "train_json": "labels/dsb2018/budget/mask_B164605_s4.json",
9
+ "gpu": "Tesla V100-SXM2-32GB",
10
+ "torch": "1.10.2",
11
+ "train_sec": 2937.9,
12
+ "best_iter": 2000,
13
+ "best_val": 61.76937977198289,
14
+ "select_metric": "AP",
15
+ "predict": {
16
+ "val": {
17
+ "n_dets": 4645,
18
+ "sec": 4.7
19
+ },
20
+ "test": {
21
+ "n_dets": 4478,
22
+ "sec": 7.4
23
+ }
24
+ }
25
+ }