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

cleemesser · Mar 5, 2020 · aa1a4b0 · aa1a4b0
1 parent 1db9675
commit aa1a4b0
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']