hlydecker commited on
Commit
be24e47
1 Parent(s): 0639dc1

feature: v2 tree model

Browse files
tree_model_weights/trees_v2_cfg.yaml ADDED
@@ -0,0 +1,325 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CUDNN_BENCHMARK: false
2
+ DATALOADER:
3
+ ASPECT_RATIO_GROUPING: true
4
+ FILTER_EMPTY_ANNOTATIONS: true
5
+ NUM_WORKERS: 2
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
+ - trees_v2_test
15
+ TRAIN:
16
+ - trees_v2_train
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: polygon
28
+ MAX_SIZE_TEST: 1333
29
+ MAX_SIZE_TRAIN: 1333
30
+ MIN_SIZE_TEST: 800
31
+ MIN_SIZE_TRAIN:
32
+ - 640
33
+ - 672
34
+ - 704
35
+ - 736
36
+ - 768
37
+ - 800
38
+ MIN_SIZE_TRAIN_SAMPLING: choice
39
+ RANDOM_FLIP: horizontal
40
+ MODEL:
41
+ ANCHOR_GENERATOR:
42
+ ANGLES:
43
+ - - -90
44
+ - 0
45
+ - 90
46
+ ASPECT_RATIOS:
47
+ - - 0.5
48
+ - 1.0
49
+ - 2.0
50
+ NAME: DefaultAnchorGenerator
51
+ OFFSET: 0.0
52
+ SIZES:
53
+ - - 32
54
+ - - 64
55
+ - - 128
56
+ - - 256
57
+ - - 512
58
+ BACKBONE:
59
+ FREEZE_AT: 2
60
+ NAME: build_resnet_fpn_backbone
61
+ DEVICE: cuda
62
+ FPN:
63
+ FUSE_TYPE: sum
64
+ IN_FEATURES:
65
+ - res2
66
+ - res3
67
+ - res4
68
+ - res5
69
+ NORM: ''
70
+ OUT_CHANNELS: 256
71
+ KEYPOINT_ON: false
72
+ LOAD_PROPOSALS: false
73
+ MASK_ON: true
74
+ META_ARCHITECTURE: GeneralizedRCNN
75
+ PANOPTIC_FPN:
76
+ COMBINE:
77
+ ENABLED: true
78
+ INSTANCES_CONFIDENCE_THRESH: 0.5
79
+ OVERLAP_THRESH: 0.5
80
+ STUFF_AREA_LIMIT: 4096
81
+ INSTANCE_LOSS_WEIGHT: 1.0
82
+ PIXEL_MEAN:
83
+ - 103.53
84
+ - 116.28
85
+ - 123.675
86
+ PIXEL_STD:
87
+ - 1.0
88
+ - 1.0
89
+ - 1.0
90
+ PROPOSAL_GENERATOR:
91
+ MIN_SIZE: 0
92
+ NAME: RPN
93
+ RESNETS:
94
+ DEFORM_MODULATED: false
95
+ DEFORM_NUM_GROUPS: 1
96
+ DEFORM_ON_PER_STAGE:
97
+ - false
98
+ - false
99
+ - false
100
+ - false
101
+ DEPTH: 101
102
+ NORM: FrozenBN
103
+ NUM_GROUPS: 1
104
+ OUT_FEATURES:
105
+ - res2
106
+ - res3
107
+ - res4
108
+ - res5
109
+ RES2_OUT_CHANNELS: 256
110
+ RES5_DILATION: 1
111
+ STEM_OUT_CHANNELS: 64
112
+ STRIDE_IN_1X1: true
113
+ WIDTH_PER_GROUP: 64
114
+ RETINANET:
115
+ BBOX_REG_LOSS_TYPE: smooth_l1
116
+ BBOX_REG_WEIGHTS: &id002
117
+ - 1.0
118
+ - 1.0
119
+ - 1.0
120
+ - 1.0
121
+ FOCAL_LOSS_ALPHA: 0.25
122
+ FOCAL_LOSS_GAMMA: 2.0
123
+ IN_FEATURES:
124
+ - p3
125
+ - p4
126
+ - p5
127
+ - p6
128
+ - p7
129
+ IOU_LABELS:
130
+ - 0
131
+ - -1
132
+ - 1
133
+ IOU_THRESHOLDS:
134
+ - 0.4
135
+ - 0.5
136
+ NMS_THRESH_TEST: 0.5
137
+ NORM: ''
138
+ NUM_CLASSES: 80
139
+ NUM_CONVS: 4
140
+ PRIOR_PROB: 0.01
141
+ SCORE_THRESH_TEST: 0.05
142
+ SMOOTH_L1_LOSS_BETA: 0.1
143
+ TOPK_CANDIDATES_TEST: 1000
144
+ ROI_BOX_CASCADE_HEAD:
145
+ BBOX_REG_WEIGHTS:
146
+ - &id001
147
+ - 10.0
148
+ - 10.0
149
+ - 5.0
150
+ - 5.0
151
+ - - 20.0
152
+ - 20.0
153
+ - 10.0
154
+ - 10.0
155
+ - - 30.0
156
+ - 30.0
157
+ - 15.0
158
+ - 15.0
159
+ IOUS:
160
+ - 0.5
161
+ - 0.6
162
+ - 0.7
163
+ ROI_BOX_HEAD:
164
+ BBOX_REG_LOSS_TYPE: smooth_l1
165
+ BBOX_REG_LOSS_WEIGHT: 1.0
166
+ BBOX_REG_WEIGHTS: *id001
167
+ CLS_AGNOSTIC_BBOX_REG: false
168
+ CONV_DIM: 256
169
+ FC_DIM: 1024
170
+ FED_LOSS_FREQ_WEIGHT_POWER: 0.5
171
+ FED_LOSS_NUM_CLASSES: 50
172
+ NAME: FastRCNNConvFCHead
173
+ NORM: ''
174
+ NUM_CONV: 0
175
+ NUM_FC: 2
176
+ POOLER_RESOLUTION: 7
177
+ POOLER_SAMPLING_RATIO: 0
178
+ POOLER_TYPE: ROIAlignV2
179
+ SMOOTH_L1_BETA: 0.0
180
+ TRAIN_ON_PRED_BOXES: false
181
+ USE_FED_LOSS: false
182
+ USE_SIGMOID_CE: 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: 2
198
+ POSITIVE_FRACTION: 0.25
199
+ PROPOSAL_APPEND_GT: true
200
+ SCORE_THRESH_TEST: 0.7
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: 1000
254
+ POST_NMS_TOPK_TRAIN: 1000
255
+ PRE_NMS_TOPK_TEST: 1000
256
+ PRE_NMS_TOPK_TRAIN: 2000
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: /home/ubuntu/ssd/trees_v2_out_06122023/model_final.pth
272
+ OUTPUT_DIR: /home/ubuntu/ssd/trees_v2_out_06122023
273
+ SEED: -1
274
+ SOLVER:
275
+ AMP:
276
+ ENABLED: false
277
+ BASE_LR: 0.00025
278
+ BASE_LR_END: 0.0
279
+ BIAS_LR_FACTOR: 1.0
280
+ CHECKPOINT_PERIOD: 5000
281
+ CLIP_GRADIENTS:
282
+ CLIP_TYPE: value
283
+ CLIP_VALUE: 1.0
284
+ ENABLED: false
285
+ NORM_TYPE: 2.0
286
+ GAMMA: 0.1
287
+ IMS_PER_BATCH: 8
288
+ LR_SCHEDULER_NAME: WarmupMultiStepLR
289
+ MAX_ITER: 50000
290
+ MOMENTUM: 0.9
291
+ NESTEROV: false
292
+ NUM_DECAYS: 3
293
+ REFERENCE_WORLD_SIZE: 0
294
+ RESCALE_INTERVAL: false
295
+ STEPS: []
296
+ WARMUP_FACTOR: 0.001
297
+ WARMUP_ITERS: 1000
298
+ WARMUP_METHOD: linear
299
+ WEIGHT_DECAY: 0.0001
300
+ WEIGHT_DECAY_BIAS: null
301
+ WEIGHT_DECAY_NORM: 0.0
302
+ TEST:
303
+ AUG:
304
+ ENABLED: false
305
+ FLIP: true
306
+ MAX_SIZE: 4000
307
+ MIN_SIZES:
308
+ - 400
309
+ - 500
310
+ - 600
311
+ - 700
312
+ - 800
313
+ - 900
314
+ - 1000
315
+ - 1100
316
+ - 1200
317
+ DETECTIONS_PER_IMAGE: 100
318
+ EVAL_PERIOD: 0
319
+ EXPECTED_RESULTS: []
320
+ KEYPOINT_OKS_SIGMAS: []
321
+ PRECISE_BN:
322
+ ENABLED: false
323
+ NUM_ITER: 200
324
+ VERSION: 2
325
+ VIS_PERIOD: 0
tree_model_weights/treev2_best.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5519b9a930020951a9ae96f6cfe619bedd30f5eaeb8545d6375033b0805f7ba4
3
+ size 252500430