kalinkov commited on
Commit
3e2bdcd
·
verified ·
1 Parent(s): a4f9450

Upload 2 files

Browse files
Files changed (2) hide show
  1. config.yaml +328 -0
  2. model.pth +3 -0
config.yaml ADDED
@@ -0,0 +1,328 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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_ds
16
+ TRAIN:
17
+ - train_ds
18
+ GLOBAL:
19
+ HACK: 1.0
20
+ INPUT:
21
+ CROP:
22
+ ENABLED: false
23
+ SIZE:
24
+ - 0.9
25
+ - 0.9
26
+ TYPE: relative_range
27
+ FORMAT: BGR
28
+ MASK_FORMAT: polygon
29
+ MAX_SIZE_TEST: 1333
30
+ MAX_SIZE_TRAIN: 1333
31
+ MIN_SIZE_TEST: 800
32
+ MIN_SIZE_TRAIN:
33
+ - 640
34
+ - 672
35
+ - 704
36
+ - 736
37
+ - 768
38
+ - 800
39
+ MIN_SIZE_TRAIN_SAMPLING: choice
40
+ RANDOM_FLIP: horizontal
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
+ - - 32
55
+ - - 64
56
+ - - 128
57
+ - - 256
58
+ - - 512
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: false
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
+ DEFORM_MODULATED: false
96
+ DEFORM_NUM_GROUPS: 1
97
+ DEFORM_ON_PER_STAGE:
98
+ - false
99
+ - false
100
+ - false
101
+ - false
102
+ DEPTH: 50
103
+ NORM: FrozenBN
104
+ NUM_GROUPS: 1
105
+ OUT_FEATURES:
106
+ - res2
107
+ - res3
108
+ - res4
109
+ - res5
110
+ RES2_OUT_CHANNELS: 256
111
+ RES5_DILATION: 1
112
+ STEM_OUT_CHANNELS: 64
113
+ STRIDE_IN_1X1: true
114
+ WIDTH_PER_GROUP: 64
115
+ RETINANET:
116
+ BBOX_REG_LOSS_TYPE: smooth_l1
117
+ BBOX_REG_WEIGHTS: &id002
118
+ - 1.0
119
+ - 1.0
120
+ - 1.0
121
+ - 1.0
122
+ FOCAL_LOSS_ALPHA: 0.25
123
+ FOCAL_LOSS_GAMMA: 2.0
124
+ IN_FEATURES:
125
+ - p3
126
+ - p4
127
+ - p5
128
+ - p6
129
+ - p7
130
+ IOU_LABELS:
131
+ - 0
132
+ - -1
133
+ - 1
134
+ IOU_THRESHOLDS:
135
+ - 0.4
136
+ - 0.5
137
+ NMS_THRESH_TEST: 0.5
138
+ NORM: ''
139
+ NUM_CLASSES: 80
140
+ NUM_CONVS: 4
141
+ PRIOR_PROB: 0.01
142
+ SCORE_THRESH_TEST: 0.05
143
+ SMOOTH_L1_LOSS_BETA: 0.1
144
+ TOPK_CANDIDATES_TEST: 1000
145
+ ROI_BOX_CASCADE_HEAD:
146
+ BBOX_REG_WEIGHTS:
147
+ - &id001
148
+ - 10.0
149
+ - 10.0
150
+ - 5.0
151
+ - 5.0
152
+ - - 20.0
153
+ - 20.0
154
+ - 10.0
155
+ - 10.0
156
+ - - 30.0
157
+ - 30.0
158
+ - 15.0
159
+ - 15.0
160
+ IOUS:
161
+ - 0.5
162
+ - 0.6
163
+ - 0.7
164
+ ROI_BOX_HEAD:
165
+ BBOX_REG_LOSS_TYPE: smooth_l1
166
+ BBOX_REG_LOSS_WEIGHT: 1.0
167
+ BBOX_REG_WEIGHTS: *id001
168
+ CLS_AGNOSTIC_BBOX_REG: false
169
+ CONV_DIM: 256
170
+ FC_DIM: 1024
171
+ FED_LOSS_FREQ_WEIGHT_POWER: 0.5
172
+ FED_LOSS_NUM_CLASSES: 50
173
+ NAME: FastRCNNConvFCHead
174
+ NORM: ''
175
+ NUM_CONV: 0
176
+ NUM_FC: 2
177
+ POOLER_RESOLUTION: 7
178
+ POOLER_SAMPLING_RATIO: 0
179
+ POOLER_TYPE: ROIAlignV2
180
+ SMOOTH_L1_BETA: 0.0
181
+ TRAIN_ON_PRED_BOXES: false
182
+ USE_FED_LOSS: false
183
+ USE_SIGMOID_CE: false
184
+ ROI_HEADS:
185
+ BATCH_SIZE_PER_IMAGE: 512
186
+ IN_FEATURES:
187
+ - p2
188
+ - p3
189
+ - p4
190
+ - p5
191
+ IOU_LABELS:
192
+ - 0
193
+ - 1
194
+ IOU_THRESHOLDS:
195
+ - 0.5
196
+ NAME: StandardROIHeads
197
+ NMS_THRESH_TEST: 0.5
198
+ NUM_CLASSES: 85
199
+ POSITIVE_FRACTION: 0.25
200
+ PROPOSAL_APPEND_GT: true
201
+ SCORE_THRESH_TEST: 0.2
202
+ ROI_KEYPOINT_HEAD:
203
+ CONV_DIMS:
204
+ - 512
205
+ - 512
206
+ - 512
207
+ - 512
208
+ - 512
209
+ - 512
210
+ - 512
211
+ - 512
212
+ LOSS_WEIGHT: 1.0
213
+ MIN_KEYPOINTS_PER_IMAGE: 1
214
+ NAME: KRCNNConvDeconvUpsampleHead
215
+ NORMALIZE_LOSS_BY_VISIBLE_KEYPOINTS: true
216
+ NUM_KEYPOINTS: 17
217
+ POOLER_RESOLUTION: 14
218
+ POOLER_SAMPLING_RATIO: 0
219
+ POOLER_TYPE: ROIAlignV2
220
+ ROI_MASK_HEAD:
221
+ CLS_AGNOSTIC_MASK: false
222
+ CONV_DIM: 256
223
+ NAME: MaskRCNNConvUpsampleHead
224
+ NORM: ''
225
+ NUM_CONV: 4
226
+ POOLER_RESOLUTION: 14
227
+ POOLER_SAMPLING_RATIO: 0
228
+ POOLER_TYPE: ROIAlignV2
229
+ RPN:
230
+ BATCH_SIZE_PER_IMAGE: 256
231
+ BBOX_REG_LOSS_TYPE: smooth_l1
232
+ BBOX_REG_LOSS_WEIGHT: 1.0
233
+ BBOX_REG_WEIGHTS: *id002
234
+ BOUNDARY_THRESH: -1
235
+ CONV_DIMS:
236
+ - -1
237
+ HEAD_NAME: StandardRPNHead
238
+ IN_FEATURES:
239
+ - p2
240
+ - p3
241
+ - p4
242
+ - p5
243
+ - p6
244
+ IOU_LABELS:
245
+ - 0
246
+ - -1
247
+ - 1
248
+ IOU_THRESHOLDS:
249
+ - 0.3
250
+ - 0.7
251
+ LOSS_WEIGHT: 1.0
252
+ NMS_THRESH: 0.7
253
+ POSITIVE_FRACTION: 0.5
254
+ POST_NMS_TOPK_TEST: 1000
255
+ POST_NMS_TOPK_TRAIN: 1000
256
+ PRE_NMS_TOPK_TEST: 1000
257
+ PRE_NMS_TOPK_TRAIN: 2000
258
+ SMOOTH_L1_BETA: 0.0
259
+ SEM_SEG_HEAD:
260
+ COMMON_STRIDE: 4
261
+ CONVS_DIM: 128
262
+ IGNORE_VALUE: 255
263
+ IN_FEATURES:
264
+ - p2
265
+ - p3
266
+ - p4
267
+ - p5
268
+ LOSS_WEIGHT: 1.0
269
+ NAME: SemSegFPNHead
270
+ NORM: GN
271
+ NUM_CLASSES: 54
272
+ WEIGHTS: /content/drive/MyDrive/diopsis_coco_split_flowers/results5/model_final.pth
273
+ OUTPUT_DIR: /content/drive/MyDrive/diopsis_coco_split_flowers/results5
274
+ SEED: -1
275
+ SOLVER:
276
+ AMP:
277
+ ENABLED: false
278
+ BASE_LR: 0.025
279
+ BASE_LR_END: 0.0
280
+ BIAS_LR_FACTOR: 1.0
281
+ CHECKPOINT_PERIOD: 300
282
+ CLIP_GRADIENTS:
283
+ CLIP_TYPE: value
284
+ CLIP_VALUE: 1.0
285
+ ENABLED: false
286
+ NORM_TYPE: 2.0
287
+ GAMMA: 0.1
288
+ IMS_PER_BATCH: 16
289
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
290
+ MAX_ITER: 3000
291
+ MOMENTUM: 0.9
292
+ NESTEROV: false
293
+ NUM_DECAYS: 3
294
+ REFERENCE_WORLD_SIZE: 0
295
+ RESCALE_INTERVAL: false
296
+ STEPS:
297
+ - 210000
298
+ - 250000
299
+ WARMUP_FACTOR: 0.001
300
+ WARMUP_ITERS: 1000
301
+ WARMUP_METHOD: linear
302
+ WEIGHT_DECAY: 0.0001
303
+ WEIGHT_DECAY_BIAS: null
304
+ WEIGHT_DECAY_NORM: 0.0
305
+ TEST:
306
+ AUG:
307
+ ENABLED: false
308
+ FLIP: true
309
+ MAX_SIZE: 4000
310
+ MIN_SIZES:
311
+ - 400
312
+ - 500
313
+ - 600
314
+ - 700
315
+ - 800
316
+ - 900
317
+ - 1000
318
+ - 1100
319
+ - 1200
320
+ DETECTIONS_PER_IMAGE: 100
321
+ EVAL_PERIOD: 300
322
+ EXPECTED_RESULTS: []
323
+ KEYPOINT_OKS_SIGMAS: []
324
+ PRECISE_BN:
325
+ ENABLED: false
326
+ NUM_ITER: 200
327
+ VERSION: 2
328
+ VIS_PERIOD: 0
model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7c8c3a4ea79c78d5073a778e46e5970ccc8b4642afb2b7b2f712beb1c3f5bc79
3
+ size 333469268