DurgaDeepak commited on
Commit
0bd515d
·
verified ·
1 Parent(s): 4cfdbcf

Update models/detection/detector.py

Browse files
Files changed (1) hide show
  1. models/detection/detector.py +3 -2
models/detection/detector.py CHANGED
@@ -28,8 +28,9 @@ class ObjectDetector:
28
  "yolov11b": "yolov11b",
29
  "rtdetr": "rtdetr"
30
  }
31
-
32
- model_key = model_key.lower() #to make model_key case-insensitive
 
33
  repo_map = {
34
  "yolov5n": ("ultralytics/yolov5", "yolov5n.pt"),
35
  "yolov8n": ("ultralytics/yolov8", "yolov8n.pt"),
 
28
  "yolov11b": "yolov11b",
29
  "rtdetr": "rtdetr"
30
  }
31
+
32
+ raw_key = model_key.lower()
33
+ model_key = alias_map.get(raw_key, raw_key) #to make model_key case-insensitive
34
  repo_map = {
35
  "yolov5n": ("ultralytics/yolov5", "yolov5n.pt"),
36
  "yolov8n": ("ultralytics/yolov8", "yolov8n.pt"),