Hugging Face's logo

ipd805
/
digit

Object Detection
Transformers
ONNX
PyTorch
English
Model card Files
xet
 Community
ipd805 commited on
Commit
b921faa
·
verified ·
1 Parent(s): 7639322

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +38 -16
config.json CHANGED
@@ -1,6 +1,38 @@
1
  {
2
  "name": "YOLOv8 Digit and Icon Detector",
3
  "framework": "onnx",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
4
  "labels": [
5
  {
6
  "name": "digit",
@@ -13,22 +45,12 @@
13
  "attributes": []
14
  }
15
  ],
16
- "interpretation_script": "yolo",
17
- "weights": "digit.onnx",
18
- "class_to_idx": {"digit": 0, "icon": 1},
19
- "model_parameters": {
20
- "confidence_threshold": 0.5,
21
- "nms_threshold": 0.4
22
  },
23
- "input_size": [640, 640],
24
- "preprocessing": {
25
- "resize": {
26
- "width": 640,
27
- "height": 640
28
- },
29
- "normalize": {
30
- "mean": [0.485, 0.456, 0.406],
31
- "std": [0.229, 0.224, 0.225]
32
- }
33
  }
34
  }
 
1
  {
2
  "name": "YOLOv8 Digit and Icon Detector",
3
  "framework": "onnx",
4
+ "spec": {
5
+ "class_set": [
6
+ {
7
+ "id": 0,
8
+ "name": "digit",
9
+ "color": "#ff0000"
10
+ },
11
+ {
12
+ "id": 1,
13
+ "name": "icon",
14
+ "color": "#00ff00"
15
+ }
16
+ ],
17
+ "input_size": [640, 640],
18
+ "preprocessing": {
19
+ "resize": {
20
+ "width": 640,
21
+ "height": 640
22
+ },
23
+ "normalize": {
24
+ "mean": [0.485, 0.456, 0.406],
25
+ "std": [0.229, 0.224, 0.225]
26
+ }
27
+ },
28
+ "postprocessing": {
29
+ "confidence_threshold": 0.5,
30
+ "nms_threshold": 0.4
31
+ }
32
+ },
33
+ "weights": "digit.onnx",
34
+ "interpretation_script": "yolo",
35
+ "model_type": "yolov8",
36
  "labels": [
37
  {
38
  "name": "digit",
 
45
  "attributes": []
46
  }
47
  ],
48
+ "id2label": {
49
+ "0": "digit",
50
+ "1": "icon"
 
 
 
51
  },
52
+ "label2id": {
53
+ "digit": 0,
54
+ "icon": 1
 
 
 
 
 
 
 
55
  }
56
  }