| { | |
| "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 | |
| } | |
| } | |