web-dev-qa-db-ja.com

pythonでVideoCapture opencvから特定のフレームを取得する

PythonのopencvでVideoCaptureライブラリを使用してビデオからすべてのフレームを継続的にフェッチする次のコードがあります。

import cv2

def frame_capture:
        cap = cv2.VideoCapture("video.mp4")
        while not cap.isOpened():
                cap = cv2.VideoCapture("video.mp4")
                cv2.waitKey(1000)
                print "Wait for the header"

        pos_frame = cap.get(cv2.cv.CV_CAP_PROP_POS_FRAMES)
        while True:
                flag, frame = cap.read()
                if flag:
                        # The frame is ready and already captured
                        cv2.imshow('video', frame)
                        pos_frame = cap.get(cv2.cv.CV_CAP_PROP_POS_FRAMES)
                        print str(pos_frame)+" frames"
                else:
                        # The next frame is not ready, so we try to read it again
                        cap.set(cv2.cv.CV_CAP_PROP_POS_FRAMES, pos_frame-1)
                        print "frame is not ready"
                        # It is better to wait for a while for the next frame to be ready
                        cv2.waitKey(1000)

                if cv2.waitKey(10) == 27:
                        break
                if cap.get(cv2.cv.CV_CAP_PROP_POS_FRAMES) == cap.get(cv2.cv.CV_CAP_PROP_FRAME_COUNT):
                        # If the number of captured frames is equal to the total number of frames,
                        # we stop
                        break

しかし、ビデオの特定のタイムスタンプの特定のフレームを取得したいと思います。

どうすればこれを達成できますか?

10
yusuf

VideoCaptureのset()関数を使用できます。

総フレーム数を計算できます。

cap = cv2.VideoCapture("video.mp4")
total_frames = cap.get(7)

ここで7はprop-Idです。詳細はこちら http://docs.opencv.org/2.4/modules/highgui/doc/reading_and_writing_images_and_video.html

その後、フレーム番号を設定できます。100番目のフレームを抽出するとします。

cap.set(1, 100)
ret, frame = cap.read()
cv2.imwrite("path_where_to_save_image", frame)
16
abhishek

これは私の最初の投稿なので、プロトコルに完全に従わなかった場合でも、私に侵入しないでください。抽出するフレームの数を設定する方法がわからない場合や、他の誰かがこの質問でこのスレッドを見つけた場合に備えて、June Wangに応答したかっただけです。

解決策は、古き良きforループです。

    vid = cv2.VideoCapture(video_path)
    for i in range(start_frame, how_many_frames_you_want):
        vid.set(1, i)
        ret, still = vid.read()
        cv2.imwrite(f'{video_path}_frame{i}.jpg', still)
0
Eric De Luna