ipd805 commited on
Commit
56a4dcb
·
verified ·
1 Parent(s): f76733d

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +14 -38
config.json CHANGED
@@ -1,41 +1,17 @@
1
  {
2
- "name": "YOLOv8 Digit and Icon Detector",
3
- "framework": "onnx",
4
- "type": "detection",
5
- "labels": [
6
- {
7
- "name": "digit",
8
- "color": "#ff0000"
9
- },
10
- {
11
- "name": "icon",
12
- "color": "#00ff00"
13
- }
14
  ],
15
- "model_attributes": [
16
- {
17
- "name": "confidence_threshold",
18
- "default_value": 0.25,
19
- "input_type": "number"
20
- },
21
- {
22
- "name": "iou_threshold",
23
- "default_value": 0.45,
24
- "input_type": "number"
25
- }
26
- ],
27
- "inputs": [
28
- {
29
- "name": "images",
30
- "shape": [1, 3, 640, 640],
31
- "layout": "NCHW"
32
- }
33
- ],
34
- "outputs": [
35
- {
36
- "name": "output0",
37
- "shape": [-1, 5]
38
- }
39
- ],
40
- "interpretation_script": "yolo_v8_onnx_detector"
41
  }
 
1
  {
2
+ "model_type": "object_detection",
3
+ "architecture": "yolov5",
4
+ "num_classes": 10,
5
+ "input_size": [640, 640],
6
+ "anchors": [
7
+ [10, 13, 16, 30, 33, 23],
8
+ [30, 61, 62, 45, 59, 119],
9
+ [116, 90, 156, 198, 373, 326]
 
 
 
 
10
  ],
11
+ "strides": [8, 16, 32],
12
+ "max_detections": 300,
13
+ "confidence_threshold": 0.25,
14
+ "nms_threshold": 0.45,
15
+ "pretrained": true,
16
+ "pretrained_weights": "best1-2.onnx",
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  }