Hugging Face's logo

bloomdata
/
faster-rcnn-resnet50

Model card Files
xet
 Community
crooi commited on
Commit
8efa51f
·
verified ·
1 Parent(s): c749fbf

Upload 2 files

Browse files
Files changed (2) hide show
  1. best_model.pth +3 -0
  2. config.yaml +316 -0
best_model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a14f334aadcb2627e5922082e4bc58df41c90f040c6951334d361d3ae4f2b0cf
3
+ size 165815973
config.yaml ADDED
@@ -0,0 +1,316 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 8
6
+ REPEAT_SQRT: true
7
+ REPEAT_THRESHOLD: 0.0
8
+ SAMPLER_TRAIN: TrainingSampler
9
+ DATASETS:
10
+ PRECOMPUTED_PROPOSAL_TOPK_TEST: 1000
11
+ PRECOMPUTED_PROPOSAL_TOPK_TRAIN: 2000
12
+ PROPOSAL_FILES_TEST: []
13
+ PROPOSAL_FILES_TRAIN: []
14
+ TEST:
15
+ - val
16
+ TRAIN:
17
+ - train
18
+ FLOAT32_PRECISION: ''
19
+ GLOBAL:
20
+ HACK: 1.0
21
+ INPUT:
22
+ CROP:
23
+ ENABLED: false
24
+ SIZE:
25
+ - 0.9
26
+ - 0.9
27
+ TYPE: relative_range
28
+ FORMAT: BGR
29
+ MASK_FORMAT: polygon
30
+ MAX_SIZE_TEST: 800
31
+ MAX_SIZE_TRAIN: 800
32
+ MIN_SIZE_TEST: 640
33
+ MIN_SIZE_TRAIN:
34
+ - 640
35
+ MIN_SIZE_TRAIN_SAMPLING: choice
36
+ RANDOM_FLIP: horizontal
37
+ MODEL:
38
+ ANCHOR_GENERATOR:
39
+ ANGLES:
40
+ - - -90
41
+ - 0
42
+ - 90
43
+ ASPECT_RATIOS:
44
+ - - 0.5
45
+ - 1.0
46
+ - 2.0
47
+ NAME: DefaultAnchorGenerator
48
+ OFFSET: 0.0
49
+ SIZES:
50
+ - - 32
51
+ - - 64
52
+ - - 128
53
+ - - 256
54
+ - - 512
55
+ BACKBONE:
56
+ FREEZE_AT: 2
57
+ NAME: build_resnet_fpn_backbone
58
+ DEVICE: cuda
59
+ FPN:
60
+ FUSE_TYPE: sum
61
+ IN_FEATURES:
62
+ - res2
63
+ - res3
64
+ - res4
65
+ - res5
66
+ NORM: ''
67
+ OUT_CHANNELS: 256
68
+ KEYPOINT_ON: false
69
+ LOAD_PROPOSALS: false
70
+ MASK_ON: false
71
+ META_ARCHITECTURE: GeneralizedRCNN
72
+ PANOPTIC_FPN:
73
+ COMBINE:
74
+ ENABLED: true
75
+ INSTANCES_CONFIDENCE_THRESH: 0.5
76
+ OVERLAP_THRESH: 0.5
77
+ STUFF_AREA_LIMIT: 4096
78
+ INSTANCE_LOSS_WEIGHT: 1.0
79
+ PIXEL_MEAN:
80
+ - 103.53
81
+ - 116.28
82
+ - 123.675
83
+ PIXEL_STD:
84
+ - 1.0
85
+ - 1.0
86
+ - 1.0
87
+ PROPOSAL_GENERATOR:
88
+ MIN_SIZE: 0
89
+ NAME: RPN
90
+ RESNETS:
91
+ DEFORM_MODULATED: false
92
+ DEFORM_NUM_GROUPS: 1
93
+ DEFORM_ON_PER_STAGE:
94
+ - false
95
+ - false
96
+ - false
97
+ - false
98
+ DEPTH: 50
99
+ NORM: FrozenBN
100
+ NUM_GROUPS: 1
101
+ OUT_FEATURES:
102
+ - res2
103
+ - res3
104
+ - res4
105
+ - res5
106
+ RES2_OUT_CHANNELS: 256
107
+ RES5_DILATION: 1
108
+ STEM_OUT_CHANNELS: 64
109
+ STRIDE_IN_1X1: true
110
+ WIDTH_PER_GROUP: 64
111
+ RETINANET:
112
+ BBOX_REG_LOSS_TYPE: smooth_l1
113
+ BBOX_REG_WEIGHTS: &id002
114
+ - 1.0
115
+ - 1.0
116
+ - 1.0
117
+ - 1.0
118
+ FOCAL_LOSS_ALPHA: 0.25
119
+ FOCAL_LOSS_GAMMA: 2.0
120
+ IN_FEATURES:
121
+ - p3
122
+ - p4
123
+ - p5
124
+ - p6
125
+ - p7
126
+ IOU_LABELS:
127
+ - 0
128
+ - -1
129
+ - 1
130
+ IOU_THRESHOLDS:
131
+ - 0.4
132
+ - 0.5
133
+ NMS_THRESH_TEST: 0.5
134
+ NORM: ''
135
+ NUM_CLASSES: 80
136
+ NUM_CONVS: 4
137
+ PRIOR_PROB: 0.01
138
+ SCORE_THRESH_TEST: 0.05
139
+ SMOOTH_L1_LOSS_BETA: 0.1
140
+ TOPK_CANDIDATES_TEST: 1000
141
+ ROI_BOX_CASCADE_HEAD:
142
+ BBOX_REG_WEIGHTS:
143
+ - &id001
144
+ - 10.0
145
+ - 10.0
146
+ - 5.0
147
+ - 5.0
148
+ - - 20.0
149
+ - 20.0
150
+ - 10.0
151
+ - 10.0
152
+ - - 30.0
153
+ - 30.0
154
+ - 15.0
155
+ - 15.0
156
+ IOUS:
157
+ - 0.5
158
+ - 0.6
159
+ - 0.7
160
+ ROI_BOX_HEAD:
161
+ BBOX_REG_LOSS_TYPE: smooth_l1
162
+ BBOX_REG_LOSS_WEIGHT: 1.0
163
+ BBOX_REG_WEIGHTS: *id001
164
+ CLS_AGNOSTIC_BBOX_REG: false
165
+ CONV_DIM: 256
166
+ FC_DIM: 1024
167
+ FED_LOSS_FREQ_WEIGHT_POWER: 0.5
168
+ FED_LOSS_NUM_CLASSES: 50
169
+ NAME: FastRCNNConvFCHead
170
+ NORM: ''
171
+ NUM_CONV: 0
172
+ NUM_FC: 2
173
+ POOLER_RESOLUTION: 7
174
+ POOLER_SAMPLING_RATIO: 0
175
+ POOLER_TYPE: ROIAlignV2
176
+ SMOOTH_L1_BETA: 0.0
177
+ TRAIN_ON_PRED_BOXES: false
178
+ USE_FED_LOSS: false
179
+ USE_SIGMOID_CE: false
180
+ ROI_HEADS:
181
+ BATCH_SIZE_PER_IMAGE: 512
182
+ IN_FEATURES:
183
+ - p2
184
+ - p3
185
+ - p4
186
+ - p5
187
+ IOU_LABELS:
188
+ - 0
189
+ - 1
190
+ IOU_THRESHOLDS:
191
+ - 0.5
192
+ NAME: StandardROIHeads
193
+ NMS_THRESH_TEST: 0.5
194
+ NUM_CLASSES: 6
195
+ POSITIVE_FRACTION: 0.25
196
+ PROPOSAL_APPEND_GT: true
197
+ SCORE_THRESH_TEST: 0.05
198
+ ROI_KEYPOINT_HEAD:
199
+ CONV_DIMS:
200
+ - 512
201
+ - 512
202
+ - 512
203
+ - 512
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ LOSS_WEIGHT: 1.0
209
+ MIN_KEYPOINTS_PER_IMAGE: 1
210
+ NAME: KRCNNConvDeconvUpsampleHead
211
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
212
+ NUM_KEYPOINTS: 17
213
+ POOLER_RESOLUTION: 14
214
+ POOLER_SAMPLING_RATIO: 0
215
+ POOLER_TYPE: ROIAlignV2
216
+ ROI_MASK_HEAD:
217
+ CLS_AGNOSTIC_MASK: false
218
+ CONV_DIM: 256
219
+ NAME: MaskRCNNConvUpsampleHead
220
+ NORM: ''
221
+ NUM_CONV: 4
222
+ POOLER_RESOLUTION: 14
223
+ POOLER_SAMPLING_RATIO: 0
224
+ POOLER_TYPE: ROIAlignV2
225
+ RPN:
226
+ BATCH_SIZE_PER_IMAGE: 256
227
+ BBOX_REG_LOSS_TYPE: smooth_l1
228
+ BBOX_REG_LOSS_WEIGHT: 1.0
229
+ BBOX_REG_WEIGHTS: *id002
230
+ BOUNDARY_THRESH: -1
231
+ CONV_DIMS:
232
+ - -1
233
+ HEAD_NAME: StandardRPNHead
234
+ IN_FEATURES:
235
+ - p2
236
+ - p3
237
+ - p4
238
+ - p5
239
+ - p6
240
+ IOU_LABELS:
241
+ - 0
242
+ - -1
243
+ - 1
244
+ IOU_THRESHOLDS:
245
+ - 0.3
246
+ - 0.7
247
+ LOSS_WEIGHT: 1.0
248
+ NMS_THRESH: 0.7
249
+ POSITIVE_FRACTION: 0.5
250
+ POST_NMS_TOPK_TEST: 1000
251
+ POST_NMS_TOPK_TRAIN: 1000
252
+ PRE_NMS_TOPK_TEST: 1000
253
+ PRE_NMS_TOPK_TRAIN: 2000
254
+ SMOOTH_L1_BETA: 0.0
255
+ SEM_SEG_HEAD:
256
+ COMMON_STRIDE: 4
257
+ CONVS_DIM: 128
258
+ IGNORE_VALUE: 255
259
+ IN_FEATURES:
260
+ - p2
261
+ - p3
262
+ - p4
263
+ - p5
264
+ LOSS_WEIGHT: 1.0
265
+ NAME: SemSegFPNHead
266
+ NORM: GN
267
+ NUM_CLASSES: 54
268
+ WEIGHTS: https://dl.fbaipublicfiles.com/detectron2/COCO-Detection/faster_rcnn_R_50_FPN_3x/137849458/model_final_280758.pkl
269
+ OUTPUT_DIR: ./outputs/training_20241026_170628
270
+ SEED: 42
271
+ SOLVER:
272
+ AMP:
273
+ ENABLED: true
274
+ BASE_LR: 0.00025
275
+ BASE_LR_END: 0.0
276
+ BIAS_LR_FACTOR: 1.0
277
+ CHECKPOINT_PERIOD: 2000
278
+ CLIP_GRADIENTS:
279
+ CLIP_TYPE: value
280
+ CLIP_VALUE: 0.5
281
+ ENABLED: true
282
+ NORM_TYPE: 2.0
283
+ GAMMA: 0.1
284
+ IMS_PER_BATCH: 16
285
+ LR_SCHEDULER_NAME: WarmupCosineLR
286
+ MAX_ITER: 40000
287
+ MOMENTUM: 0.9
288
+ NESTEROV: false
289
+ NUM_DECAYS: 3
290
+ REFERENCE_WORLD_SIZE: 0
291
+ RESCALE_INTERVAL: false
292
+ STEPS:
293
+ - 210000
294
+ - 250000
295
+ WARMUP_FACTOR: 0.001
296
+ WARMUP_ITERS: 4000
297
+ WARMUP_METHOD: linear
298
+ WEIGHT_DECAY: 0.0001
299
+ WEIGHT_DECAY_BIAS: null
300
+ WEIGHT_DECAY_NORM: 0.0
301
+ TEST:
302
+ AUG:
303
+ ENABLED: false
304
+ FLIP: true
305
+ MAX_SIZE: 800
306
+ MIN_SIZES:
307
+ - 640
308
+ DETECTIONS_PER_IMAGE: 100
309
+ EVAL_PERIOD: 500
310
+ EXPECTED_RESULTS: []
311
+ KEYPOINT_OKS_SIGMAS: []
312
+ PRECISE_BN:
313
+ ENABLED: false
314
+ NUM_ITER: 200
315
+ VERSION: 2
316
+ VIS_PERIOD: 0