wuhp commited on
Commit
953b4d3
Β·
verified Β·
1 Parent(s): a6cb66e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -10
app.py CHANGED
@@ -31,16 +31,18 @@ from rfdetr.util.coco_classes import COCO_CLASSES
31
  ###############################################################################
32
 
33
  YOLO_MODEL_MAP: Dict[str, str] = {
34
- "YOLOv12‑n": "yolov12n.pt",
35
- "YOLOv12‑s": "yolov12s.pt",
36
- "YOLOv12‑m": "yolov12m.pt",
37
- "YOLOv12‑l": "yolov12l.pt",
38
- "YOLOv12‑x": "yolov12x.pt",
39
- "YOLOv11‑n": "yolov11n.pt",
40
- "YOLOv11‑s": "yolov11s.pt",
41
- "YOLOv11‑m": "yolov11m.pt",
42
- "YOLOv11‑l": "yolov11l.pt",
43
- "YOLOv11‑x": "yolov11x.pt",
 
 
44
  }
45
 
46
  RFDETR_MODEL_MAP = {
 
31
  ###############################################################################
32
 
33
  YOLO_MODEL_MAP: Dict[str, str] = {
34
+ # NOTE: Ultralytics filenames do NOT include the "v" character
35
+ "YOLOv12‑n": "yolo12n.pt",
36
+ "YOLOv12‑s": "yolo12s.pt",
37
+ "YOLOv12‑m": "yolo12m.pt",
38
+ "YOLOv12‑l": "yolo12l.pt",
39
+ "YOLOv12‑x": "yolo12x.pt",
40
+ "YOLOv11‑n": "yolo11n.pt",
41
+ "YOLOv11‑s": "yolo11s.pt",
42
+ "YOLOv11‑m": "yolo11m.pt",
43
+ "YOLOv11‑l": "yolo11l.pt",
44
+ "YOLOv11‑x": "yolo11x.pt",
45
+ }
46
  }
47
 
48
  RFDETR_MODEL_MAP = {