try alternative implementation to make it faster

This commit is contained in:
SkalskiP committed 2023-06-28 20:02:03 +02:00
1 parent cd929525f6
commit 17020b23de
1 file changed
+5 -2
+5 -2
View File
@@ -139,9 +139,12 @@ def get_video_frames_generator(
success, frame = video.read()
if not success or frame_position >= end:
break
frame_position += stride
video.set(cv2.CAP_PROP_POS_FRAMES, frame_position)
yield frame
for _ in range(stride - 1):
success = video.grab()
if not success:
break
frame_position += stride
video.release()