Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
35 |
-
"YOLOv12β
|
36 |
-
"YOLOv12β
|
37 |
-
"YOLOv12β
|
38 |
-
"YOLOv12β
|
39 |
-
"
|
40 |
-
"YOLOv11β
|
41 |
-
"YOLOv11β
|
42 |
-
"YOLOv11β
|
43 |
-
"YOLOv11β
|
|
|
|
|
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 = {
|