add robust JSON handeling for auto annotation runner (#758)

cvat-ai · Oct 8, 2019 · a18fb6d · a18fb6d
1 parent a18e83b
commit a18fb6d
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/utils/auto_annotation/run_model.py b/utils/auto_annotation/run_model.py
@@ -70,7 +70,11 @@ def main():
         return
 
     with open(mapping_file) as json_file:
-        mapping = json.load(json_file)
+        try:
+            mapping = json.load(json_file)
+        except json.decoder.JSONDecodeError:
+            logging.critical('JSON file not able to be parsed! Check file')
+            return
 
     try:
         mapping = mapping['label_map']