ipd805 commited on
Commit
92ad35b
·
verified ·
1 Parent(s): 7ec8c14

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +5 -14
config.json CHANGED
@@ -1,26 +1,17 @@
1
  {
2
- "model_type": "yolov8",
3
  "framework": "onnx",
4
  "task": "object-detection",
5
  "architecture": "YOLOv8",
6
  "format": "ONNX",
7
  "num_labels": 2,
8
- "id2label": {
9
- "0": "digit",
10
- "1": "icon"
11
- },
12
- "label2id": {
13
- "digit": 0,
14
- "icon": 1
15
- },
16
  "input_shape": [1, 3, 640, 640],
17
  "inference_framework": "onnxruntime",
18
  "preprocessor": {
19
  "type": "YOLOv8ImageProcessor",
20
- "size": {
21
- "height": 640,
22
- "width": 640
23
- },
24
  "do_normalize": true,
25
  "do_resize": true,
26
  "resize_method": "letterbox"
@@ -31,4 +22,4 @@
31
  "iou_threshold": 0.45,
32
  "max_detections_per_image": 300
33
  }
34
- }
 
1
  {
2
+ "model_type": "object-detection",
3
  "framework": "onnx",
4
  "task": "object-detection",
5
  "architecture": "YOLOv8",
6
  "format": "ONNX",
7
  "num_labels": 2,
8
+ "id2label": { "0": "digit", "1": "icon" },
9
+ "label2id": { "digit": 0, "icon": 1 },
 
 
 
 
 
 
10
  "input_shape": [1, 3, 640, 640],
11
  "inference_framework": "onnxruntime",
12
  "preprocessor": {
13
  "type": "YOLOv8ImageProcessor",
14
+ "size": { "height": 640, "width": 640 },
 
 
 
15
  "do_normalize": true,
16
  "do_resize": true,
17
  "resize_method": "letterbox"
 
22
  "iou_threshold": 0.45,
23
  "max_detections_per_image": 300
24
  }
25
+ }