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

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +13 -15
config.json CHANGED
@@ -1,17 +1,15 @@
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
  }
 
1
  {
2
+ "name": "YOLOv8 Digit and Icon Detector",
3
+ "framework": "onnx",
4
+ "task_type": "object-detection",
5
+ "architecture": "onnx",
6
+ "model_type": "onnx",
7
+ "weights": "best1-2.onnx",
8
+ "labels": [
9
+ {
10
+ "name": "digit",
11
+ "0": "digit",
12
+ "1": "icon"
13
+ },
14
+ "interpretation_script": "onnx"
 
 
15
  }