Update config.json
Browse files- config.json +7 -23
config.json
CHANGED
|
@@ -1,17 +1,8 @@
|
|
| 1 |
{
|
| 2 |
-
"
|
| 3 |
-
"framework": "
|
| 4 |
"task_type": "object-detection",
|
| 5 |
-
"architecture": "
|
| 6 |
-
"model_type": "yolov8",
|
| 7 |
-
"weights": "best1-2.onnx",
|
| 8 |
-
"config": {
|
| 9 |
-
"num_classes": 2,
|
| 10 |
-
"input_size": [640, 640],
|
| 11 |
-
"anchors": "auto",
|
| 12 |
-
"strides": [8, 16, 32],
|
| 13 |
-
"class_names": ["digit", "icon"]
|
| 14 |
-
},
|
| 15 |
"labels": [
|
| 16 |
{
|
| 17 |
"id": 0,
|
|
@@ -24,14 +15,8 @@
|
|
| 24 |
"color": "#00ff00"
|
| 25 |
}
|
| 26 |
],
|
| 27 |
-
"
|
| 28 |
-
|
| 29 |
-
"1": "icon"
|
| 30 |
-
},
|
| 31 |
-
"label2id": {
|
| 32 |
-
"digit": 0,
|
| 33 |
-
"icon": 1
|
| 34 |
-
},
|
| 35 |
"preprocessing": {
|
| 36 |
"resize_mode": "letterbox",
|
| 37 |
"size": [640, 640],
|
|
@@ -45,9 +30,8 @@
|
|
| 45 |
"max_detections": 300
|
| 46 |
},
|
| 47 |
"metadata": {
|
| 48 |
-
"description": "
|
| 49 |
"author": "Your Name",
|
| 50 |
-
"version": "1.0"
|
| 51 |
-
"license": "MIT"
|
| 52 |
}
|
| 53 |
}
|
|
|
|
| 1 |
{
|
| 2 |
+
"model_name": "Digit and Icon Detector",
|
| 3 |
+
"framework": "onnx",
|
| 4 |
"task_type": "object-detection",
|
| 5 |
+
"architecture": "ONNX",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 6 |
"labels": [
|
| 7 |
{
|
| 8 |
"id": 0,
|
|
|
|
| 15 |
"color": "#00ff00"
|
| 16 |
}
|
| 17 |
],
|
| 18 |
+
"input_shape": [1, 3, 640, 640],
|
| 19 |
+
"input_format": "NCHW",
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 20 |
"preprocessing": {
|
| 21 |
"resize_mode": "letterbox",
|
| 22 |
"size": [640, 640],
|
|
|
|
| 30 |
"max_detections": 300
|
| 31 |
},
|
| 32 |
"metadata": {
|
| 33 |
+
"description": "ONNX model for detecting digits and icons",
|
| 34 |
"author": "Your Name",
|
| 35 |
+
"version": "1.0"
|
|
|
|
| 36 |
}
|
| 37 |
}
|