From f1ac6ecccc296ff54eb9bf8a7787ef542165adad Mon Sep 17 00:00:00 2001 From: HonzaCuhel Date: Thu, 31 Oct 2024 16:26:55 +0100 Subject: [PATCH] Fix: V9 detection --- tools/utils/version_detection.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tools/utils/version_detection.py b/tools/utils/version_detection.py index 9d4cc5c..3fa1276 100644 --- 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