ipd805 commited on
Commit
50a4e02
·
verified ·
1 Parent(s): da9ca8b

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +7 -23
config.json CHANGED
@@ -1,17 +1,8 @@
1
  {
2
- "name": "YOLOv8 Digit and Icon Detector",
3
- "framework": "yolov8",
4
  "task_type": "object-detection",
5
- "architecture": "YOLOv8",
6
- "model_type": "yolov8",
7
- "weights": "best1-2.onnx",
8
- "config": {
9
- "num_classes": 2,
10
- "input_size": [640, 640],
11
- "anchors": "auto",
12
- "strides": [8, 16, 32],
13
- "class_names": ["digit", "icon"]
14
- },
15
  "labels": [
16
  {
17
  "id": 0,
@@ -24,14 +15,8 @@
24
  "color": "#00ff00"
25
  }
26
  ],
27
- "id2label": {
28
- "0": "digit",
29
- "1": "icon"
30
- },
31
- "label2id": {
32
- "digit": 0,
33
- "icon": 1
34
- },
35
  "preprocessing": {
36
  "resize_mode": "letterbox",
37
  "size": [640, 640],
@@ -45,9 +30,8 @@
45
  "max_detections": 300
46
  },
47
  "metadata": {
48
- "description": "YOLOv8 model for detecting digits and icons",
49
  "author": "Your Name",
50
- "version": "1.0",
51
- "license": "MIT"
52
  }
53
  }
 
1
  {
2
+ "model_name": "Digit and Icon Detector",
3
+ "framework": "onnx",
4
  "task_type": "object-detection",
5
+ "architecture": "ONNX",
 
 
 
 
 
 
 
 
 
6
  "labels": [
7
  {
8
  "id": 0,
 
15
  "color": "#00ff00"
16
  }
17
  ],
18
+ "input_shape": [1, 3, 640, 640],
19
+ "input_format": "NCHW",
 
 
 
 
 
 
20
  "preprocessing": {
21
  "resize_mode": "letterbox",
22
  "size": [640, 640],
 
30
  "max_detections": 300
31
  },
32
  "metadata": {
33
+ "description": "ONNX model for detecting digits and icons",
34
  "author": "Your Name",
35
+ "version": "1.0"
 
36
  }
37
  }