Skip to content

Commit

Permalink
extract yolo deploy
Browse files Browse the repository at this point in the history
  • Loading branch information
lrosemberg committed Dec 16, 2024
1 parent 594bff7 commit d4d60e7
Showing 1 changed file with 10 additions and 0 deletions.
10 changes: 10 additions & 0 deletions roboflow/core/version.py
Original file line number Diff line number Diff line change
Expand Up @@ -521,6 +521,16 @@ def deploy(self, model_type: str, model_path: str, filename: str = "weights/best
self.deploy_yolonas(model_type, model_path, filename)
return

self.deploy_yolo(model_type, model_path, filename)

def deploy_yolo(self, model_type: str, model_path: str, filename: str) -> None:
"""Deploy YOLO model to Roboflow.
Args:
model_type (str): The type of YOLO model to deploy
model_path (str): Path to model weights
filename (str): Name of weights file
"""
if "yolov8" in model_type:
try:
import torch
Expand Down

0 comments on commit d4d60e7

Please sign in to comment.