Upload Yolov6ForObjectDetection

config.json

@@ -2,6 +2,7 @@
   "architectures": [
     "Yolov6ForObjectDetection"
   ],
+  "atss_warmup_epoch": 0,
   "backbone_csp_e": 0.0,
   "backbone_cspsppf": true,
   "backbone_fuse_P2": true,
@@ -23,6 +24,7 @@
   "block_type": "Yolov6RepVGGBlock",
   "class_loss_coefficient": 1.0,
   "dfl_loss_coefficient": 1.0,
+  "export": false,
   "forward_fuse": false,
   "head_anchors": 3,
   "head_in_channels": [
@@ -36,7 +38,6 @@
     16,
     32
   ],
-  "hidden_size": 192,
   "id2label": {
     "0": "person",
     "1": "bicycle",
@@ -119,13 +120,9 @@
     "78": "hair drier",
     "79": "toothbrush"
   },
-  "image_size": [
-    800,
-    1333
-  ],
+  "image_size": 640,
   "in_channels": 3,
   "initializer_range": 0.02,
-  "intermediate_size": 768,
   "iou_loss_coefficient": 1.0,
   "iou_type": "siou",
   "label2id": {
@@ -227,12 +224,9 @@
     128
   ],
   "neck_stage_block_type": "Yolov6RepBlock",
-  "num_attention_heads": 3,
-  "num_hidden_layers": 12,
   "reg_max": 0,
   "reg_max_proj": 16,
   "torch_dtype": "float32",
-  "transformers_version": "4.37.0.dev0",
-  "use_dfl": false,
-  "use_mid_position_embeddings": false
+  "transformers_version": "4.38.0.dev0",
+  "use_dfl": false
 }