Use yolov8

baernhaeckt · Aug 24, 2024 · 94971db · 94971db
1 parent 96e37a2
commit 94971db
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 1 deletion.
diff --git a/model/yolov10n.pt b/model/yolov10n.pt
diff --git a/model/yolov8n.pt b/model/yolov8n.pt
diff --git a/services/object_detection_service.py b/services/object_detection_service.py
@@ -5,7 +5,7 @@
 
 class ObjectDetectionService:
     def __init__(self):
-        self.object_detection_model = YOLO("./../model/yolov10n.pt")
+        self.object_detection_model = YOLO("./../model/yolov8n.pt")
 
     @staticmethod
     def _distance_calculator(box, img: np.ndarray) -> float: