Update config.json
Browse files- config.json +5 -14
config.json
CHANGED
|
@@ -1,26 +1,17 @@
|
|
| 1 |
{
|
| 2 |
-
"model_type": "
|
| 3 |
"framework": "onnx",
|
| 4 |
"task": "object-detection",
|
| 5 |
"architecture": "YOLOv8",
|
| 6 |
"format": "ONNX",
|
| 7 |
"num_labels": 2,
|
| 8 |
-
"id2label": {
|
| 9 |
-
|
| 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 |
+
}
|