fix crash if none frame returned (#203)

Co-authored-by: gferraro <g.ferraro22@gmail.com>
TheCacophonyProject · May 19, 2024 · c8155e2 · c8155e2
1 parent 4d33643
commit c8155e2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/ml_tools/interpreter.py b/src/ml_tools/interpreter.py
@@ -300,6 +300,9 @@ def preprocess_segments(
             for frame_index in frame_indices:
                 region = track.bounds_history[frame_index - track.start_frame]
                 f = clip.get_frame(region.frame_number)
+                if f is None:
+                    logging.warn("Could not get frame {}", region.frame_number)
+                    continue
                 if region.blank or region.width <= 0 or region.height <= 0:
                     continue