| |
| |
|
|
|
|
| REGISTERED_ALIASES = { |
| "yolov8n-640": "coco/3", |
| "yolov8n-1280": "coco/9", |
| "yolov8s-640": "coco/6", |
| "yolov8s-1280": "coco/10", |
| "yolov8m-640": "coco/8", |
| "yolov8m-1280": "coco/11", |
| "yolov8l-640": "coco/7", |
| "yolov8l-1280": "coco/12", |
| "yolov8x-640": "coco/5", |
| "yolov8x-1280": "coco/13", |
| "yolo-nas-s-640": "coco/14", |
| "yolo-nas-m-640": "coco/15", |
| "yolo-nas-l-640": "coco/16", |
| "yolov8n-seg-640": "coco-dataset-vdnr1/2", |
| "yolov8n-seg-1280": "coco-dataset-vdnr1/7", |
| "yolov8s-seg-640": "coco-dataset-vdnr1/4", |
| "yolov8s-seg-1280": "coco-dataset-vdnr1/8", |
| "yolov8m-seg-640": "coco-dataset-vdnr1/5", |
| "yolov8m-seg-1280": "coco-dataset-vdnr1/9", |
| "yolov8l-seg-640": "coco-dataset-vdnr1/6", |
| "yolov8l-seg-1280": "coco-dataset-vdnr1/10", |
| "yolov8x-seg-640": "coco-dataset-vdnr1/3", |
| "yolov8x-seg-1280": "coco-dataset-vdnr1/11", |
| } |
|
|
|
|
| def resolve_roboflow_model_alias(model_id: str) -> str: |
| return REGISTERED_ALIASES.get(model_id, model_id) |
|
|
