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 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): 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: