Pythonで動画のキャプチャについて

Question

指で秒数を数えている動画を撮って、キャプチャで約５秒間の動画から５枚の画像を取得したいと考えています。（１枚目：指が１本の画像、２枚目：２本の画像、３枚目：３本の画像...）

# -*- coding: utf-8 -*-
import cv2
import os
#動画ファイルを読み込む
file_name = u"test3.mov"
cap = cv2.VideoCapture(file_name)
#スクリーンキャプチャを保存するディレクトリ
dir_name = "screen_caps"
if not os.path.exists(dir_name):
    os.mkdir(dir_name)
#フレーム数を取得
frame_count = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
fps = int(cap.get(cv2.CAP_PROP_FPS))

# 総フレーム数とFPSを確認
print("FRAME_COUNT: ",  frame_count)
print("FPS: ", fps )

start_pos = 0

for i in range(start_pos, round(frame_count), round(fps)):
    _, frame = cap.read()
    cv2.imwrite(dir_name+ "/" + str(i) + ".png", frame)

以上のように、1秒ごとにキャプチャしていくコードを作成しました。
しかし、5枚とも指が１本の画像となってしまいます。

for i in range(start_pos, round(frame_count)):
    _, frame = cap.read()
    cv2.imwrite(dir_name+ "/" + str(i) + ".png", frame)

ronund(fps)をなくすと、うまくフレームごとに画像を取得しているので指が増えていきます。
原因はround(fps)でしょうか。
よろしくお願いいたします。

Comments

[LouiS0616]

ループ前にフレーム数とfpsを出力しているようですが、どのような値になりましたか。

[gymgym]

フレーム数が189、fpsが30となっています

Answer 1

cap.readを使うと、愚直に次のフレームを読みに行ってしまいます。
次のように空読みすれば動くかと思います。

for i in range(frame_count):
    if i % fps != 0:
        cap.read()
        continue

    ret, image = cap.read()
    if not ret:
        break

    cv2.imwrite(f'{dir_name}/{i}.png', frame)

空読み部分は次のように書いても可。

for _ in range(fps):
    cap.read()

あるいは

プロパティCAP_PROP_POS_MSECを書き換えても良さそう。
ですが、OpenCVのプロパティが無視される環境があるので、なんとなく避けちゃいますね。

now_pos = 0
for i in range(5):
    now_pos += 1000
    cap.set(
        cv2.CAP_PROP_POS_MSEC, now_pos
    )
    
    ret, frame = cap.read()
    if not ret:
        break

    cv2.imwrite(f'{dir_name}/{i}.png', frame)