| { |
| "model_type": "pytorch", |
| "framework": "onnx", |
| "task": "object-detection", |
| "architecture": "YOLOv8", |
| "format": "onnx", |
| "num_labels": 2, |
| "id2label": { "0": "digit", "1": "icon" }, |
| "label2id": { "digit": 0, "icon": 1 }, |
| "input_shape": [1, 3, 640, 640], |
| "inference_framework": "onnxruntime", |
| "preprocessor": { |
| "type": "YOLOv8ImageProcessor", |
| "size": { "height": 640, "width": 640 }, |
| "do_normalize": true, |
| "do_resize": true, |
| "resize_method": "letterbox" |
| }, |
| "postprocessor": { |
| "type": "YOLOv8Postprocessor", |
| "confidence_threshold": 0.25, |
| "iou_threshold": 0.45, |
| "max_detections_per_image": 300 |
| } |
| } |
|
|
