sbchoi commited on
Commit
d3f1ca0
1 Parent(s): 58c37ef

Upload Yolov6ForObjectDetection

Browse files
Files changed (1) hide show
  1. config.json +4 -4
config.json CHANGED
@@ -23,7 +23,7 @@
23
  "backbone_stage_block_type": "Yolov6RepBlock",
24
  "block_type": "Yolov6RepVGGBlock",
25
  "class_loss_coefficient": 1.0,
26
- "dfl_loss_coefficient": 1.0,
27
  "export": false,
28
  "forward_fuse": false,
29
  "head_anchors": 3,
@@ -123,7 +123,7 @@
123
  "image_size": 640,
124
  "in_channels": 3,
125
  "initializer_range": 0.02,
126
- "iou_loss_coefficient": 1.0,
127
  "iou_type": "siou",
128
  "label2id": {
129
  "aeroplane": 4,
@@ -207,7 +207,7 @@
207
  "wine glass": 40,
208
  "zebra": 22
209
  },
210
- "model_type": "yolov6n",
211
  "neck_csp_e": 0.0,
212
  "neck_num_repeats": [
213
  4,
@@ -227,6 +227,6 @@
227
  "reg_max": 0,
228
  "reg_max_proj": 16,
229
  "torch_dtype": "float32",
230
- "transformers_version": "4.38.0.dev0",
231
  "use_dfl": false
232
  }
 
23
  "backbone_stage_block_type": "Yolov6RepBlock",
24
  "block_type": "Yolov6RepVGGBlock",
25
  "class_loss_coefficient": 1.0,
26
+ "dfl_loss_coefficient": 0.5,
27
  "export": false,
28
  "forward_fuse": false,
29
  "head_anchors": 3,
 
123
  "image_size": 640,
124
  "in_channels": 3,
125
  "initializer_range": 0.02,
126
+ "iou_loss_coefficient": 2.5,
127
  "iou_type": "siou",
128
  "label2id": {
129
  "aeroplane": 4,
 
207
  "wine glass": 40,
208
  "zebra": 22
209
  },
210
+ "model_type": "yolov6",
211
  "neck_csp_e": 0.0,
212
  "neck_num_repeats": [
213
  4,
 
227
  "reg_max": 0,
228
  "reg_max_proj": 16,
229
  "torch_dtype": "float32",
230
+ "transformers_version": "4.42.0.dev0",
231
  "use_dfl": false
232
  }