Upload Yolov6ForObjectDetection

config.json

@@ -23,7 +23,7 @@
   "backbone_stage_block_type": "Yolov6RepBlock",
   "block_type": "Yolov6RepVGGBlock",
   "class_loss_coefficient": 1.0,
-  "dfl_loss_coefficient": 1.0,
+  "dfl_loss_coefficient": 0.5,
   "export": false,
   "forward_fuse": false,
   "head_anchors": 3,
@@ -123,7 +123,7 @@
   "image_size": 640,
   "in_channels": 3,
   "initializer_range": 0.02,
-  "iou_loss_coefficient": 1.0,
+  "iou_loss_coefficient": 2.5,
   "iou_type": "siou",
   "label2id": {
     "aeroplane": 4,
@@ -207,7 +207,7 @@
     "wine glass": 40,
     "zebra": 22
   },
-  "model_type": "yolov6n",
+  "model_type": "yolov6",
   "neck_csp_e": 0.0,
   "neck_num_repeats": [
     4,
@@ -227,6 +227,6 @@
   "reg_max": 0,
   "reg_max_proj": 16,
   "torch_dtype": "float32",
-  "transformers_version": "4.38.0.dev0",
+  "transformers_version": "4.42.0.dev0",
   "use_dfl": false
 }