ipd805 commited on
Commit
2ac4e1c
·
verified ·
1 Parent(s): 15cd432

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +14 -4
config.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
  "name": "YOLOv8 Digit and Icon Detector",
3
  "framework": "onnx",
4
- "task_type": "image-classification",
5
- "architecture": "onnx",
6
- "model_type": "onnx",
7
  "weights": "best1-2.onnx",
8
  "labels": [
9
  {
@@ -19,5 +19,15 @@
19
  "0": "digit",
20
  "1": "icon"
21
  },
22
- "interpretation_script": "onnx"
 
 
 
 
 
 
 
 
 
 
23
  }
 
1
  {
2
  "name": "YOLOv8 Digit and Icon Detector",
3
  "framework": "onnx",
4
+ "task_type": "detection",
5
+ "architecture": "yolov8",
6
+ "model_type": "yolov8",
7
  "weights": "best1-2.onnx",
8
  "labels": [
9
  {
 
19
  "0": "digit",
20
  "1": "icon"
21
  },
22
+ "label2id": {
23
+ "digit": 0,
24
+ "icon": 1
25
+ },
26
+ "interpretation_script": "yolo",
27
+ "config": {
28
+ "input_size": [640, 640],
29
+ "confidence_threshold": 0.25,
30
+ "iou_threshold": 0.45,
31
+ "max_detections": 300
32
+ }
33
  }