Update config.json

config.json

@@ -1,25 +1,53 @@
 {
-  "model_name": "ipd805/digit",
-  "model_type": "ObjectDetection",
-  "framework": "onnx",
-  "class_set": {
-    "name": "digit-icon-classes",
-    "classes": [
-      {
-        "name": "digit",
-        "id": 0,
-        "color": "#ff0000"
-      },
-      {
-        "name": "icon",
-        "id": 1,
-        "color": "#00ff00"
-      }
-    ]
+  "name": "YOLOv8 Digit and Icon Detector",
+  "framework": "yolov8",
+  "task_type": "object-detection",
+  "architecture": "YOLOv8",
+  "model_type": "yolov8",
+  "weights": "best1-2.onnx",
+  "config": {
+    "num_classes": 2,
+    "input_size": [640, 640],
+    "anchors": "auto",
+    "strides": [8, 16, 32],
+    "class_names": ["digit", "icon"]
   },
-  "input_shape": [640, 640, 3],
-  "input_type": "image",
-  "output_type": "tensor",
-  "confidence_threshold": 0.5,
-  "iou_threshold": 0.5
+  "labels": [
+    {
+      "id": 0,
+      "name": "digit",
+      "color": "#ff0000"
+    },
+    {
+      "id": 1,
+      "name": "icon",
+      "color": "#00ff00"
+    }
+  ],
+  "id2label": {
+    "0": "digit",
+    "1": "icon"
+  },
+  "label2id": {
+    "digit": 0,
+    "icon": 1
+  },
+  "preprocessing": {
+    "resize_mode": "letterbox",
+    "size": [640, 640],
+    "mean": [0, 0, 0],
+    "std": [255, 255, 255],
+    "to_rgb": true
+  },
+  "postprocessing": {
+    "confidence_threshold": 0.25,
+    "iou_threshold": 0.45,
+    "max_detections": 300
+  },
+  "metadata": {
+    "description": "YOLOv8 model for detecting digits and icons",
+    "author": "Your Name",
+    "version": "1.0",
+    "license": "MIT"
+  }
 }