Update config.json

config.json

@@ -1,5 +1,19 @@
-{"input_size": 720, "task": "object-detection", "ultralyticsplus_version": "0.0.23", "ultralytics_version": "8.0.21", "model_type": "yolos",
-"id2label": {
+{
+  "architectures": [
+    "YolosForObjectDetection"
+  ],
+  "attention_probs_dropout_prob": 0.0,
+  "auxiliary_loss": false,
+  "bbox_cost": 5,
+  "bbox_loss_coefficient": 5,
+  "class_cost": 1,
+  "eos_coefficient": 0.1,
+  "giou_cost": 2,
+  "giou_loss_coefficient": 2,
+  "hidden_act": "gelu",
+  "hidden_dropout_prob": 0.0,
+  "hidden_size": 192,
+  "id2label": {
         "0": "car",
         "1": "parked_car",
         "2": "truck",
@@ -54,7 +68,13 @@
         "51": "green_traffic_light",
         "52": "cone"
     },
-    "label2id": {
+  "image_size": [
+    800,
+    1333
+  ],
+  "initializer_range": 0.02,
+  "intermediate_size": 768,
+  "label2id": {
         "car": 0,
         "parked_car": 1,
         "truck": 2,
@@ -108,5 +128,17 @@
         "yellow_traffic_light": 50,
         "green_traffic_light": 51,
         "cone": 52
-    }
+    },
+  "layer_norm_eps": 1e-12,
+  "model_type": "yolos",
+  "num_attention_heads": 3,
+  "num_channels": 3,
+  "num_detection_tokens": 100,
+  "num_hidden_layers": 12,
+  "patch_size": 16,
+  "qkv_bias": true,
+  "torch_dtype": "float32",
+  "transformers_version": "4.19.0.dev0",
+  "use_mid_position_embeddings": false
 }
+