Fix: V9 detection

luxonis · Oct 31, 2024 · f1ac6ec · f1ac6ec
1 parent bbadd23
commit f1ac6ec
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tools/utils/version_detection.py b/tools/utils/version_detection.py
@@ -51,7 +51,9 @@ def detect_version(path: str, debug: bool = False) -> str:
                 return YOLOV11_CONVERSION
             elif "yolov10" in content or "v10DetectLoss" in content:
                 return YOLOV10_CONVERSION
-            elif "yolov9" in content or ("v9-model" and "ultralytics" in content):
+            elif "yolov9" in content or (
+                "v9-model" in content and "ultralytics" in content
+            ):
                 return YOLOV9_CONVERSION
             elif (
                 "YOLOv5u" in content