Skip to content

Commit b8a4c85

Browse files
committed
Added support for yolov7-seg deploys (currently identical to yolov5 deploys)
1 parent 9246b1a commit b8a4c85

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

roboflow/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
from roboflow.core.project import Project
99
from roboflow.core.workspace import Workspace
1010

11-
__version__ = "0.2.29"
11+
__version__ = "0.2.30"
1212

1313

1414
def check_key(api_key, model, notebook, num_retries=0):

roboflow/core/version.py

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

314-
supported_models = ["yolov8", "yolov5"]
314+
supported_models = ["yolov8", "yolov5", "yolov7-seg"]
315315

316316
if model_type not in supported_models:
317317
raise (
@@ -334,7 +334,7 @@ def deploy(self, model_type: str, model_path: str) -> None:
334334
[("ultralytics", "<=", "8.0.20")]
335335
)
336336

337-
elif model_type == "yolov5":
337+
elif model_type in ["yolov5", "yolov7-seg"]:
338338
try:
339339
import torch
340340
except ImportError as e:
@@ -378,7 +378,7 @@ def deploy(self, model_type: str, model_path: str) -> None:
378378
"ultralytics_version": ultralytics.__version__,
379379
"model_type": model_type,
380380
}
381-
elif model_type == "yolov5":
381+
elif model_type in ["yolov5", "yolov7-seg"]:
382382
# parse from yaml for yolov5
383383

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

0 commit comments

Comments
 (0)