fix --step argument

2022-12-08 15:06:24 +00:00 · 2022-12-08 15:06:24 +00:00 · eab4633c55
commit eab4633c55
parent 72e7c1ec6f
1 changed files with 2 additions and 1 deletions
--- a/scripts/preprocess/extract_video.py
+++ b/scripts/preprocess/extract_video.py
@ -32,7 +32,8 @@ def extract_video(videoname, path, start, end, step):
        if cnt < start:continue
        if cnt >= end:break
        if not ret:continue
-        cv2.imwrite(join(outpath, '{:06d}.jpg'.format(cnt)), frame)
+        if (cnt % step ==0):
+            cv2.imwrite(join(outpath, '{:06d}.jpg'.format(cnt)), frame)
    video.release()
    return base