DurgaDeepak commited on
Commit
b9a2f92
·
verified ·
1 Parent(s): 79acde6

Update models/detection/detector.py

Browse files
Files changed (1) hide show
  1. models/detection/detector.py +1 -0
models/detection/detector.py CHANGED
@@ -35,6 +35,7 @@ class ObjectDetector:
35
  model_key = alias_map.get(raw_key, raw_key) #to make model_key case-insensitive
36
  repo_map = {
37
  "yolov5n": ("ultralytics/yolov5", "yolov5n.pt"),
 
38
  "yolov8n": ("ultralytics/yolov8", "yolov8n.pt"),
39
  "yolov8s": ("ultralytics/yolov8", "yolov8s.pt"),
40
  "yolov8l": ("ultralytics/yolov8", "yolov8l.pt"),
 
35
  model_key = alias_map.get(raw_key, raw_key) #to make model_key case-insensitive
36
  repo_map = {
37
  "yolov5n": ("ultralytics/yolov5", "yolov5n.pt"),
38
+ "yolov5s": ("ultralytics/yolov5", "yolov5s.pt"),
39
  "yolov8n": ("ultralytics/yolov8", "yolov8n.pt"),
40
  "yolov8s": ("ultralytics/yolov8", "yolov8s.pt"),
41
  "yolov8l": ("ultralytics/yolov8", "yolov8l.pt"),