Updated the constant variables for Yolov8 (#917)

Co-authored-by: fatih <[email protected]>
obss · Nov 5, 2023 · 5e36667 · 5e36667
1 parent 978d2ee
commit 5e36667
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/sahi/utils/yolov8.py b/sahi/utils/yolov8.py
@@ -14,11 +14,11 @@ class Yolov8TestConstants:
     YOLOV8M_MODEL_URL = "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8m.pt"
     YOLOV8M_MODEL_PATH = "tests/data/models/yolov8/yolov8m.pt"
 
-    YOLOV8M_MODEL_URL = "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8l.pt"
-    YOLOV8M_MODEL_PATH = "tests/data/models/yolov8/yolov8l.pt"
+    YOLOV8L_MODEL_URL = "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8l.pt"
+    YOLOV8L_MODEL_PATH = "tests/data/models/yolov8/yolov8l.pt"
 
-    YOLOV8M_MODEL_URL = "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8x.pt"
-    YOLOV8M_MODEL_PATH = "tests/data/models/yolov8/yolov8x.pt"
+    YOLOV8X_MODEL_URL = "https://github.com/ultralytics/assets/releases/download/v0.0.0/yolov8x.pt"
+    YOLOV8X_MODEL_PATH = "tests/data/models/yolov8/yolov8x.pt"
 
 
 def download_yolov8n_model(destination_path: Optional[str] = None):