Zhen Ye commited on
Commit
38a60a8
·
1 Parent(s): 29e1c2b

Update Lite model to use spencercdz/YOLOv8m_defence

Browse files
Files changed (1) hide show
  1. models/detectors/yolov8.py +2 -2
models/detectors/yolov8.py CHANGED
@@ -12,8 +12,8 @@ from models.detectors.base import DetectionResult, ObjectDetector
12
  class HuggingFaceYoloV8Detector(ObjectDetector):
13
  """YOLOv8 detector whose weights are fetched from the Hugging Face Hub."""
14
 
15
- REPO_ID = "Ultralytics/YOLOv8"
16
- WEIGHT_FILE = "yolov8s.pt"
17
  supports_batch = True
18
  max_batch_size = 8
19
 
 
12
  class HuggingFaceYoloV8Detector(ObjectDetector):
13
  """YOLOv8 detector whose weights are fetched from the Hugging Face Hub."""
14
 
15
+ REPO_ID = "spencercdz/YOLOv8m_defence"
16
+ WEIGHT_FILE = "yolov8m_defence.pt"
17
  supports_batch = True
18
  max_batch_size = 8
19