Skip to content

Commit 56b8c9c

Browse files
committed
support old ultralytics deploys
1 parent 36f185b commit 56b8c9c

File tree

1 file changed

+27
-13
lines changed

1 file changed

+27
-13
lines changed

roboflow/core/version.py

Lines changed: 27 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -284,7 +284,7 @@ def train(self, speed=None, checkpoint=None) -> bool:
284284

285285
return True
286286

287-
def deploy(self, model_type: str, model_path: str) -> None:
287+
def deploy(self, model_type: str, model_path: str) -> None:
288288
"""Uploads provided weights file to Roboflow
289289
290290
Args:
@@ -318,18 +318,32 @@ def deploy(self, model_type: str, model_path: str) -> None:
318318
class_names.sort(key=lambda x: x[0])
319319
class_names = [x[1] for x in class_names]
320320

321-
model_artifacts = {
322-
"names": class_names,
323-
"yaml": model["model"].yaml,
324-
"nc": model["model"].nc,
325-
"args": {
326-
k: val
327-
for k, val in model["model"].args.__dict__.items()
328-
if k != "hydra"
329-
},
330-
"ultralytics_version": ultralytics.__version__,
331-
"model_type": model_type,
332-
}
321+
try:
322+
model_artifacts = {
323+
"names": class_names,
324+
"yaml": model["model"].yaml,
325+
"nc": model["model"].nc,
326+
"args": {
327+
k: val
328+
for k, val in model["model"].args.items()
329+
if ((k == "model") or (k == "imgsz") or (k == "batch"))
330+
},
331+
"ultralytics_version": ultralytics.__version__,
332+
"model_type": model_type,
333+
}
334+
except:
335+
model_artifacts = {
336+
"names": class_names,
337+
"yaml": model["model"].yaml,
338+
"nc": model["model"].nc,
339+
"args": {
340+
k: val
341+
for k, val in model["model"].args.__dict__.items()
342+
if ((k == "model") or (k == "imgsz") or (k == "batch"))
343+
},
344+
"ultralytics_version": ultralytics.__version__,
345+
"model_type": model_type,
346+
}
333347

334348
with open(model_path + "model_artifacts.json", "w") as fp:
335349
json.dump(model_artifacts, fp)

0 commit comments

Comments
 (0)