Merge pull request #243 from yqtl/extract_video_fix

fix --step argument
2023-02-11 20:54:34 +08:00 · 2023-02-11 20:54:34 +08:00 · e287c6bd81
commit e287c6bd81
parent 7a80646902 eab4633c55
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