make detector not stop when failing on a single video

breunigs · Jan 27, 2024 · cd1dafe · cd1dafe
1 parent 3d27748
commit cd1dafe
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tools/detection/detector.py b/tools/detection/detector.py
@@ -119,7 +119,12 @@ def process(item, model, outer_bar):
         target=lambda: detections.update(load_json_gzip(wip)))
     gzip_loader.start()
 
-    frames, frame_count = load_video(name)
+    try:
+        frames, frame_count = load_video(name)
+    except:
+        print(f"\nfailed to load video \"{name}\", skipping")
+        return gzip_loader
+
     gzip_loader.join()
 
     bytes_per_frame = math.floor(size / float(frame_count))