Merge pull request #231 from roboflow/feature/yolov9-upload

YOLOv9 Deploy
roboflow · Feb 27, 2024 · bd5f01a · bd5f01a
2 parents ef6e3d9 + cb455d6
commit bd5f01a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/roboflow/__init__.py b/roboflow/__init__.py
@@ -14,7 +14,7 @@
 from roboflow.models import CLIPModel, GazeModel  # noqa: F401
 from roboflow.util.general import write_line
 
-__version__ = "1.1.20"
+__version__ = "1.1.21"
 
 
 def check_key(api_key, model, notebook, num_retries=0):

diff --git a/roboflow/core/version.py b/roboflow/core/version.py
@@ -425,7 +425,7 @@ def deploy(self, model_type: str, model_path: str) -> None:
             model_path (str): File path to model weights to be uploaded
         """
 
-        supported_models = ["yolov5", "yolov7-seg", "yolov8"]
+        supported_models = ["yolov5", "yolov7-seg", "yolov8", "yolov9"]
 
         if not any(supported_model in model_type for supported_model in supported_models):
             raise (ValueError(f"Model type {model_type} not supported. Supported models are" f" {supported_models}"))
@@ -443,7 +443,7 @@ def deploy(self, model_type: str, model_path: str) -> None:
 
             print_warn_for_wrong_dependencies_versions([("ultralytics", "==", "8.0.196")], ask_to_continue=True)
 
-        elif "yolov5" in model_type or "yolov7" in model_type:
+        elif "yolov5" in model_type or "yolov7" in model_type or "yolov9" in model_type:
             try:
                 import torch
             except ImportError:
@@ -493,7 +493,7 @@ def deploy(self, model_type: str, model_path: str) -> None:
                     "ultralytics_version": ultralytics.__version__,
                     "model_type": model_type,
                 }
-        elif "yolov5" in model_type or "yolov7" in model_type:
+        elif "yolov5" in model_type or "yolov7" in model_type or "yolov9" in model_type:
             # parse from yaml for yolov5
 
             with open(os.path.join(model_path, "opt.yaml"), "r") as stream: