ultralytics.nn.tasks.YOLOv10DetectionModel does not expose 'nc' property

roboflow · Aug 26, 2024 · 1fb6471 · 1fb6471
1 parent c28cb01
commit 1fb6471
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/roboflow/core/version.py b/roboflow/core/version.py
@@ -532,7 +532,7 @@ def deploy(self, model_type: str, model_path: str, filename: str = "weights/best
 
         if "yolov8" in model_type or "yolov10" in model_type:
             # try except for backwards compatibility with older versions of ultralytics
-            if "-cls" in model_type:
+            if "-cls" in model_type or model_type.startswith("yolov10"):
                 nc = model["model"].yaml["nc"]
                 args = model["train_args"]
             else: