Merge pull request #98 from roboflow/supportOldUltralytics

Support Old Versions of Ultralytics YOLOv8 Deploy

Author: Jacobsolawetz

roboflow/__init__.py

@@ -8,7 +8,7 @@
 from roboflow.core.project import Project
 from roboflow.core.workspace import Workspace
 
-__version__ = "0.2.26"
+__version__ = "0.2.27"
 
 
 def check_key(api_key, model, notebook, num_retries=0):

roboflow/core/version.py

@@ -318,18 +318,32 @@ def deploy(self, model_type: str, model_path: str) -> None:
             class_names.sort(key=lambda x: x[0])
             class_names = [x[1] for x in class_names]
 
-            model_artifacts = {
-                "names": class_names,
-                "yaml": model["model"].yaml,
-                "nc": model["model"].nc,
-                "args": {
-                    k: val
-                    for k, val in model["model"].args.__dict__.items()
-                    if k != "hydra"
-                },
-                "ultralytics_version": ultralytics.__version__,
-                "model_type": model_type,
-            }
+            try:
+                model_artifacts = {
+                    "names": class_names,
+                    "yaml": model["model"].yaml,
+                    "nc": model["model"].nc,
+                    "args": {
+                        k: val
+                        for k, val in model["model"].args.items()
+                        if ((k == "model") or (k == "imgsz") or (k == "batch"))
+                    },
+                    "ultralytics_version": ultralytics.__version__,
+                    "model_type": model_type,
+                }
+            except:
+                model_artifacts = {
+                    "names": class_names,
+                    "yaml": model["model"].yaml,
+                    "nc": model["model"].nc,
+                    "args": {
+                        k: val
+                        for k, val in model["model"].args.__dict__.items()
+                        if ((k == "model") or (k == "imgsz") or (k == "batch"))
+                    },
+                    "ultralytics_version": ultralytics.__version__,
+                    "model_type": model_type,
+                }
 
             with open(model_path + "model_artifacts.json", "w") as fp:
                 json.dump(model_artifacts, fp)