diff --git a/modules/capturer.py b/modules/capturer.py index 1dee22d..5ef149a 100644 --- a/modules/capturer.py +++ b/modules/capturer.py @@ -14,8 +14,13 @@ def get_video_frame(video_path: str, frame_number: int = 0) -> Any: if modules.globals.color_correction: capture.set(cv2.CAP_PROP_CONVERT_RGB, 1) - frame_total = capture.get(cv2.CAP_PROP_FRAME_COUNT) - capture.set(cv2.CAP_PROP_POS_FRAMES, min(frame_total, frame_number - 1)) + frame_total = int(capture.get(cv2.CAP_PROP_FRAME_COUNT)) + if frame_total <= 0: + capture.release() + return None + + target_index = 0 if frame_number <= 1 else min(frame_total - 1, frame_number - 1) + capture.set(cv2.CAP_PROP_POS_FRAMES, target_index) has_frame, frame = capture.read() if has_frame and modules.globals.color_correction: