Skip to content

Commit 667cc49

Browse files
committed
Added yolov9 and yolov9-seg as model type options for deploy
1 parent ef6e3d9 commit 667cc49

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

roboflow/core/version.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -425,7 +425,7 @@ def deploy(self, model_type: str, model_path: str) -> None:
425425
model_path (str): File path to model weights to be uploaded
426426
"""
427427

428-
supported_models = ["yolov5", "yolov7-seg", "yolov8"]
428+
supported_models = ["yolov5", "yolov7-seg", "yolov8", "yolov9", "yolov9-seg"]
429429

430430
if not any(supported_model in model_type for supported_model in supported_models):
431431
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:
443443

444444
print_warn_for_wrong_dependencies_versions([("ultralytics", "==", "8.0.196")], ask_to_continue=True)
445445

446-
elif "yolov5" in model_type or "yolov7" in model_type:
446+
elif "yolov5" in model_type or "yolov7" in model_type or "yolov9" in model_type:
447447
try:
448448
import torch
449449
except ImportError:
@@ -493,7 +493,7 @@ def deploy(self, model_type: str, model_path: str) -> None:
493493
"ultralytics_version": ultralytics.__version__,
494494
"model_type": model_type,
495495
}
496-
elif "yolov5" in model_type or "yolov7" in model_type:
496+
elif "yolov5" in model_type or "yolov7" in model_type or "yolov9" in model_type:
497497
# parse from yaml for yolov5
498498

499499
with open(os.path.join(model_path, "opt.yaml"), "r") as stream:

0 commit comments

Comments
 (0)