Recover max frames to run inference (WIP)

SainsburyWellcomeCentre · Nov 12, 2024 · 585e445 · 585e445
1 parent e371bf9
commit 585e445
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/crabs/tracker/track_video.py b/crabs/tracker/track_video.py
@@ -232,6 +232,13 @@ def core_detection_and_tracking(self):
         # Loop over frames
         frame_idx = 0
         while input_video_object.isOpened():
+            # Break if beyond end frame (mostly for debugging)
+            if (
+                self.args.max_frames_to_read
+                and frame_idx + 1 > self.args.max_frames_to_read
+            ):
+                break
+
             # Read frame
             ret, frame = input_video_object.read()
             if not ret: