Hugging Face's logo

unstructuredio
/
oer-checkbox

Model card Files Community
qued commited on
Commit
4c2e4b6
•
1 Parent(s): 13943ee

Upload detectron2_oer_checkbox.json

Browse files
Files changed (1) hide show
  1. detectron2_oer_checkbox.json +402 -0
detectron2_oer_checkbox.json ADDED
@@ -0,0 +1,402 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "VERSION": 2,
3
+ "MODEL": {
4
+ "LOAD_PROPOSALS": false,
5
+ "MASK_ON": false,
6
+ "KEYPOINT_ON": false,
7
+ "DEVICE": "cpu",
8
+ "META_ARCHITECTURE": "GeneralizedRCNN",
9
+ "WEIGHTS": "https://www.dropbox.com/s/dgy9c10wykk4lq4/model_final.pth?dl=1",
10
+ "PIXEL_MEAN": [
11
+ 103.53,
12
+ 116.28,
13
+ 123.675
14
+ ],
15
+ "PIXEL_STD": [
16
+ 1.0,
17
+ 1.0,
18
+ 1.0
19
+ ],
20
+ "BACKBONE": {
21
+ "NAME": "build_resnet_fpn_backbone",
22
+ "FREEZE_AT": 2
23
+ },
24
+ "FPN": {
25
+ "IN_FEATURES": [
26
+ "res2",
27
+ "res3",
28
+ "res4",
29
+ "res5"
30
+ ],
31
+ "OUT_CHANNELS": 256,
32
+ "NORM": "",
33
+ "FUSE_TYPE": "sum"
34
+ },
35
+ "PROPOSAL_GENERATOR": {
36
+ "NAME": "RPN",
37
+ "MIN_SIZE": 0
38
+ },
39
+ "ANCHOR_GENERATOR": {
40
+ "NAME": "DefaultAnchorGenerator",
41
+ "SIZES": [
42
+ [
43
+ 32
44
+ ],
45
+ [
46
+ 64
47
+ ],
48
+ [
49
+ 128
50
+ ],
51
+ [
52
+ 256
53
+ ],
54
+ [
55
+ 512
56
+ ]
57
+ ],
58
+ "ASPECT_RATIOS": [
59
+ [
60
+ 0.5,
61
+ 1.0,
62
+ 2.0
63
+ ]
64
+ ],
65
+ "ANGLES": [
66
+ [
67
+ -90,
68
+ 0,
69
+ 90
70
+ ]
71
+ ],
72
+ "OFFSET": 0.0
73
+ },
74
+ "RPN": {
75
+ "HEAD_NAME": "StandardRPNHead",
76
+ "IN_FEATURES": [
77
+ "p2",
78
+ "p3",
79
+ "p4",
80
+ "p5",
81
+ "p6"
82
+ ],
83
+ "BOUNDARY_THRESH": -1,
84
+ "IOU_THRESHOLDS": [
85
+ 0.3,
86
+ 0.7
87
+ ],
88
+ "IOU_LABELS": [
89
+ 0,
90
+ -1,
91
+ 1
92
+ ],
93
+ "BATCH_SIZE_PER_IMAGE": 256,
94
+ "POSITIVE_FRACTION": 0.5,
95
+ "BBOX_REG_LOSS_TYPE": "smooth_l1",
96
+ "BBOX_REG_LOSS_WEIGHT": 1.0,
97
+ "BBOX_REG_WEIGHTS": [
98
+ 1.0,
99
+ 1.0,
100
+ 1.0,
101
+ 1.0
102
+ ],
103
+ "SMOOTH_L1_BETA": 0.0,
104
+ "LOSS_WEIGHT": 1.0,
105
+ "PRE_NMS_TOPK_TRAIN": 2000,
106
+ "PRE_NMS_TOPK_TEST": 1000,
107
+ "POST_NMS_TOPK_TRAIN": 1000,
108
+ "POST_NMS_TOPK_TEST": 1000,
109
+ "NMS_THRESH": 0.7,
110
+ "CONV_DIMS": [
111
+ -1
112
+ ]
113
+ },
114
+ "ROI_HEADS": {
115
+ "NAME": "StandardROIHeads",
116
+ "NUM_CLASSES": 2,
117
+ "IN_FEATURES": [
118
+ "p2",
119
+ "p3",
120
+ "p4",
121
+ "p5"
122
+ ],
123
+ "IOU_THRESHOLDS": [
124
+ 0.5
125
+ ],
126
+ "IOU_LABELS": [
127
+ 0,
128
+ 1
129
+ ],
130
+ "BATCH_SIZE_PER_IMAGE": 128,
131
+ "POSITIVE_FRACTION": 0.25,
132
+ "SCORE_THRESH_TEST": 0.8,
133
+ "NMS_THRESH_TEST": 0.5,
134
+ "PROPOSAL_APPEND_GT": true
135
+ },
136
+ "ROI_BOX_HEAD": {
137
+ "NAME": "FastRCNNConvFCHead",
138
+ "BBOX_REG_LOSS_TYPE": "smooth_l1",
139
+ "BBOX_REG_LOSS_WEIGHT": 1.0,
140
+ "BBOX_REG_WEIGHTS": [
141
+ 10.0,
142
+ 10.0,
143
+ 5.0,
144
+ 5.0
145
+ ],
146
+ "SMOOTH_L1_BETA": 0.0,
147
+ "POOLER_RESOLUTION": 7,
148
+ "POOLER_SAMPLING_RATIO": 0,
149
+ "POOLER_TYPE": "ROIAlignV2",
150
+ "NUM_FC": 2,
151
+ "FC_DIM": 1024,
152
+ "NUM_CONV": 0,
153
+ "CONV_DIM": 256,
154
+ "NORM": "",
155
+ "CLS_AGNOSTIC_BBOX_REG": false,
156
+ "TRAIN_ON_PRED_BOXES": false
157
+ },
158
+ "ROI_BOX_CASCADE_HEAD": {
159
+ "BBOX_REG_WEIGHTS": [
160
+ [
161
+ 10.0,
162
+ 10.0,
163
+ 5.0,
164
+ 5.0
165
+ ],
166
+ [
167
+ 20.0,
168
+ 20.0,
169
+ 10.0,
170
+ 10.0
171
+ ],
172
+ [
173
+ 30.0,
174
+ 30.0,
175
+ 15.0,
176
+ 15.0
177
+ ]
178
+ ],
179
+ "IOUS": [
180
+ 0.5,
181
+ 0.6,
182
+ 0.7
183
+ ]
184
+ },
185
+ "ROI_MASK_HEAD": {
186
+ "NAME": "MaskRCNNConvUpsampleHead",
187
+ "POOLER_RESOLUTION": 14,
188
+ "POOLER_SAMPLING_RATIO": 0,
189
+ "NUM_CONV": 4,
190
+ "CONV_DIM": 256,
191
+ "NORM": "",
192
+ "CLS_AGNOSTIC_MASK": false,
193
+ "POOLER_TYPE": "ROIAlignV2"
194
+ },
195
+ "ROI_KEYPOINT_HEAD": {
196
+ "NAME": "KRCNNConvDeconvUpsampleHead",
197
+ "POOLER_RESOLUTION": 14,
198
+ "POOLER_SAMPLING_RATIO": 0,
199
+ "CONV_DIMS": [
200
+ 512,
201
+ 512,
202
+ 512,
203
+ 512,
204
+ 512,
205
+ 512,
206
+ 512,
207
+ 512
208
+ ],
209
+ "NUM_KEYPOINTS": 17,
210
+ "MIN_KEYPOINTS_PER_IMAGE": 1,
211
+ "NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS": true,
212
+ "LOSS_WEIGHT": 1.0,
213
+ "POOLER_TYPE": "ROIAlignV2"
214
+ },
215
+ "SEM_SEG_HEAD": {
216
+ "NAME": "SemSegFPNHead",
217
+ "IN_FEATURES": [
218
+ "p2",
219
+ "p3",
220
+ "p4",
221
+ "p5"
222
+ ],
223
+ "IGNORE_VALUE": 255,
224
+ "NUM_CLASSES": 54,
225
+ "CONVS_DIM": 128,
226
+ "COMMON_STRIDE": 4,
227
+ "NORM": "GN",
228
+ "LOSS_WEIGHT": 1.0
229
+ },
230
+ "PANOPTIC_FPN": {
231
+ "INSTANCE_LOSS_WEIGHT": 1.0,
232
+ "COMBINE": {
233
+ "ENABLED": true,
234
+ "OVERLAP_THRESH": 0.5,
235
+ "STUFF_AREA_LIMIT": 4096,
236
+ "INSTANCES_CONFIDENCE_THRESH": 0.5
237
+ }
238
+ },
239
+ "RETINANET": {
240
+ "NUM_CLASSES": 80,
241
+ "IN_FEATURES": [
242
+ "p3",
243
+ "p4",
244
+ "p5",
245
+ "p6",
246
+ "p7"
247
+ ],
248
+ "NUM_CONVS": 4,
249
+ "IOU_THRESHOLDS": [
250
+ 0.4,
251
+ 0.5
252
+ ],
253
+ "IOU_LABELS": [
254
+ 0,
255
+ -1,
256
+ 1
257
+ ],
258
+ "PRIOR_PROB": 0.01,
259
+ "SCORE_THRESH_TEST": 0.05,
260
+ "TOPK_CANDIDATES_TEST": 1000,
261
+ "NMS_THRESH_TEST": 0.5,
262
+ "BBOX_REG_WEIGHTS": [
263
+ 1.0,
264
+ 1.0,
265
+ 1.0,
266
+ 1.0
267
+ ],
268
+ "FOCAL_LOSS_GAMMA": 2.0,
269
+ "FOCAL_LOSS_ALPHA": 0.25,
270
+ "SMOOTH_L1_LOSS_BETA": 0.1,
271
+ "BBOX_REG_LOSS_TYPE": "smooth_l1",
272
+ "NORM": ""
273
+ },
274
+ "RESNETS": {
275
+ "DEPTH": 50,
276
+ "OUT_FEATURES": [
277
+ "res2",
278
+ "res3",
279
+ "res4",
280
+ "res5"
281
+ ],
282
+ "NUM_GROUPS": 1,
283
+ "NORM": "FrozenBN",
284
+ "WIDTH_PER_GROUP": 64,
285
+ "STRIDE_IN_1X1": true,
286
+ "RES5_DILATION": 1,
287
+ "RES2_OUT_CHANNELS": 256,
288
+ "STEM_OUT_CHANNELS": 64,
289
+ "DEFORM_ON_PER_STAGE": [
290
+ false,
291
+ false,
292
+ false,
293
+ false
294
+ ],
295
+ "DEFORM_MODULATED": false,
296
+ "DEFORM_NUM_GROUPS": 1
297
+ }
298
+ },
299
+ "INPUT": {
300
+ "MIN_SIZE_TRAIN": [
301
+ 640,
302
+ 672,
303
+ 704,
304
+ 736,
305
+ 768,
306
+ 800
307
+ ],
308
+ "MIN_SIZE_TRAIN_SAMPLING": "choice",
309
+ "MAX_SIZE_TRAIN": 1333,
310
+ "MIN_SIZE_TEST": 800,
311
+ "MAX_SIZE_TEST": 1333,
312
+ "RANDOM_FLIP": "none",
313
+ "CROP": {
314
+ "ENABLED": true,
315
+ "TYPE": "relative_range",
316
+ "SIZE": [
317
+ 0.9,
318
+ 0.9
319
+ ]
320
+ },
321
+ "FORMAT": "BGR",
322
+ "MASK_FORMAT": "polygon"
323
+ },
324
+ "DATASETS": {
325
+ "TRAIN": [
326
+ "oer_checkbox_train"
327
+ ],
328
+ "PROPOSAL_FILES_TRAIN": [],
329
+ "PRECOMPUTED_PROPOSAL_TOPK_TRAIN": 2000,
330
+ "TEST": [],
331
+ "PROPOSAL_FILES_TEST": [],
332
+ "PRECOMPUTED_PROPOSAL_TOPK_TEST": 1000
333
+ },
334
+ "DATALOADER": {
335
+ "NUM_WORKERS": 2,
336
+ "ASPECT_RATIO_GROUPING": true,
337
+ "SAMPLER_TRAIN": "TrainingSampler",
338
+ "REPEAT_THRESHOLD": 0.0,
339
+ "FILTER_EMPTY_ANNOTATIONS": true
340
+ },
341
+ "SOLVER": {
342
+ "LR_SCHEDULER_NAME": "WarmupMultiStepLR",
343
+ "MAX_ITER": 600,
344
+ "BASE_LR": 0.00025,
345
+ "MOMENTUM": 0.9,
346
+ "NESTEROV": false,
347
+ "WEIGHT_DECAY": 0.0001,
348
+ "WEIGHT_DECAY_NORM": 0.0,
349
+ "GAMMA": 0.1,
350
+ "STEPS": [],
351
+ "WARMUP_FACTOR": 0.001,
352
+ "WARMUP_ITERS": 1000,
353
+ "WARMUP_METHOD": "linear",
354
+ "CHECKPOINT_PERIOD": 5000,
355
+ "IMS_PER_BATCH": 2,
356
+ "REFERENCE_WORLD_SIZE": 0,
357
+ "BIAS_LR_FACTOR": 1.0,
358
+ "WEIGHT_DECAY_BIAS": 0.0001,
359
+ "CLIP_GRADIENTS": {
360
+ "ENABLED": false,
361
+ "CLIP_TYPE": "value",
362
+ "CLIP_VALUE": 1.0,
363
+ "NORM_TYPE": 2.0
364
+ },
365
+ "AMP": {
366
+ "ENABLED": false
367
+ }
368
+ },
369
+ "TEST": {
370
+ "EXPECTED_RESULTS": [],
371
+ "EVAL_PERIOD": 0,
372
+ "KEYPOINT_OKS_SIGMAS": [],
373
+ "DETECTIONS_PER_IMAGE": 100,
374
+ "AUG": {
375
+ "ENABLED": false,
376
+ "MIN_SIZES": [
377
+ 400,
378
+ 500,
379
+ 600,
380
+ 700,
381
+ 800,
382
+ 900,
383
+ 1000,
384
+ 1100,
385
+ 1200
386
+ ],
387
+ "MAX_SIZE": 4000,
388
+ "FLIP": true
389
+ },
390
+ "PRECISE_BN": {
391
+ "ENABLED": false,
392
+ "NUM_ITER": 200
393
+ }
394
+ },
395
+ "OUTPUT_DIR": "output/oer_checkbox/faster_rcnn_R_50_FPN_3x/",
396
+ "SEED": -1,
397
+ "CUDNN_BENCHMARK": false,
398
+ "VIS_PERIOD": 0,
399
+ "GLOBAL": {
400
+ "HACK": 1.0
401
+ }
402
+ }