From 1fb64715a2090ad36b95d5731ef8168bd2166e04 Mon Sep 17 00:00:00 2001 From: Grzegorz Klimaszewski <166530809+grzegorz-roboflow@users.noreply.github.com> Date: Mon, 26 Aug 2024 16:54:57 +0200 Subject: [PATCH 1/3] ultralytics.nn.tasks.YOLOv10DetectionModel does not expose 'nc' property --- roboflow/core/version.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/roboflow/core/version.py b/roboflow/core/version.py index 0723d2be..42c24dce 100644 --- 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: From 13366599083ebba9de7c3afee32500e93b5f8c13 Mon Sep 17 00:00:00 2001 From: Grzegorz Klimaszewski <166530809+grzegorz-roboflow@users.noreply.github.com> Date: Mon, 26 Aug 2024 16:55:58 +0200 Subject: [PATCH 2/3] Add hint on how to get ultralytics with YOLOv10 models support --- README.md | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/README.md b/README.md index f29e704e..31518ca9 100644 --- a/README.md +++ b/README.md @@ -131,6 +131,11 @@ workspace.upload_dataset( # upload model weights version.deploy(model_type="yolov8", model_path=f”{HOME}/runs/detect/train/”) +# upload model weights - yolov10 +# Before attempting to upload YOLOv10 models install ultralytics like this: +# pip install git+https://github.com/THU-MIG/yolov10.git +version.deploy(model_type="yolov10", model_path=f”{HOME}/runs/detect/train/”, filename="weights.pt") + # run inference model = version.model From 53b94d57ca4730c941b5b670255ecccfd36a1c39 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Tony=20L=C3=A2mpada?= <218821+tonylampada@users.noreply.github.com> Date: Mon, 26 Aug 2024 12:10:39 -0300 Subject: [PATCH 3/3] bump version --- roboflow/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/roboflow/__init__.py b/roboflow/__init__.py index 036a7814..6aa41104 100644 --- a/roboflow/__init__.py +++ b/roboflow/__init__.py @@ -15,7 +15,7 @@ from roboflow.models import CLIPModel, GazeModel # noqa: F401 from roboflow.util.general import write_line -__version__ = "1.1.42" +__version__ = "1.1.43" def check_key(api_key, model, notebook, num_retries=0):