From 17020b23de06685b4eecbd7aab9303e44f2fafd5 Mon Sep 17 00:00:00 2001 From: SkalskiP Date: Wed, 28 Jun 2023 20:02:03 +0200 Subject: [PATCH] try alternative implementation to make it faster --- supervision/utils/video.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/supervision/utils/video.py b/supervision/utils/video.py index 4bbca10f..0ed14a4e 100644 --- a/supervision/utils/video.py +++ b/supervision/utils/video.py @@ -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()