Skip JSON files predicted by models

healthonrails · May 25, 2024 · 78776d6 · 78776d6
1 parent fbd7ad0
commit 78776d6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/annolid/utils/files.py b/annolid/utils/files.py
@@ -236,7 +236,9 @@ def merge_annotation_folders(
 
     for img in imgs:
         label_json = img.replace('png', 'json')
-        assert os.path.exists(label_json)
+        # skip json files predicted by models
+        if not os.path.exists(label_json):
+            continue
         dest_json = os.path.basename(label_json).replace(' ', '_')
         dest_img = os.path.basename(img).replace(' ', '_')
         shutil.copy(img, os.path.join(dest_dir, dest_img))