Update models/detection/detector.py

models/detection/detector.py

@@ -20,6 +20,15 @@ class ObjectDetector:
             weights_dir (str): Directory to store/download model weights.
             device (str): Inference device ("cpu" or "cuda").
         """
+        alias_map = {
+            "yolov5n-seg": "yolov5n",
+            "yolov5s-seg": "yolov5s",
+            "yolov8s": "yolov8s",
+            "yolov8l": "yolov8l",
+            "yolov11b": "yolov11b",
+            "rtdetr": "rtdetr"
+        }
+        
         model_key = model_key.lower() #to make model_key case-insensitive
         repo_map = {
             "yolov5n": ("ultralytics/yolov5", "yolov5n.pt"),