Update config.json
Browse files- config.json +14 -38
config.json
CHANGED
|
@@ -1,41 +1,17 @@
|
|
| 1 |
{
|
| 2 |
-
"
|
| 3 |
-
"
|
| 4 |
-
"
|
| 5 |
-
"
|
| 6 |
-
|
| 7 |
-
|
| 8 |
-
|
| 9 |
-
|
| 10 |
-
{
|
| 11 |
-
"name": "icon",
|
| 12 |
-
"color": "#00ff00"
|
| 13 |
-
}
|
| 14 |
],
|
| 15 |
-
"
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 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 |
}
|