Merge pull request #166 from vietanhdev/fix/crash_on_load_model

Fix for loading model configs

.gitignore

@@ -9,4 +9,5 @@ anylabeling/data
 /wheels_dist
 *_ui.py
 anylabeling/app_info.py-*
-zipped_models/*
+zipped_models/*
+venv/

.vscode/launch.json

@@ -0,0 +1,23 @@
+{
+  // Use IntelliSense to learn about possible attributes.
+  // Hover to view descriptions of existing attributes.
+  // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+  "version": "0.2.0",
+  "configurations": [
+    {
+      "name": "Debug Python Module",
+      "type": "python",
+      "request": "launch",
+      "module": "anylabeling.app",
+      "justMyCode": true
+    },
+    {
+      "name": "Profile Python Module",
+      "type": "python",
+      "request": "launch",
+      "module": "cProfile",
+      "args": ["-o", "profile_output.prof", "-m", "anylabeling.app"],
+      "justMyCode": true
+    }
+  ]
+}

anylabeling/services/auto_labeling/model_manager.py

@@ -385,6 +385,10 @@ def _load_model(self, model_id):
         # Download and extract model
         if not model_config.get("has_downloaded", True):
             model_config = self._download_and_extract_model(model_config)
+            if model_config is None:
+                return
+
+            self.model_configs[model_id].update(model_config)
 
         if model_config["type"] == "yolov5":
             from .yolov5 import YOLOv5

requirements-dev.txt

@@ -1,3 +1,4 @@
 -r requirements.txt
 build
 twine
+black

scripts/format_code.sh

@@ -1 +1 @@
-black -l 79 --exclude="tests/|anylabeling/resources/resources.py" .
+black -l 79 --exclude="tests/|anylabeling/resources/resources.py|venv/" .