在Python中使用OpenCV从VideoCapture获取特定帧

Question

我有以下代码，使用OpenCV中的VideoCapture库在Python中连续获取视频中的所有帧：

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

但是我想在视频的特定时间戳中抓取特定帧。

我该如何实现？

- yusuf

---

使用 CV_CAP_PROP_POS_MSEC 设置位置，然后获取帧。 - Miki

你能给一个小例子吗？ :) - yusuf

不是 Python，我无法测试它 :D - Miki

我认为它应该类似于：frame = cap.get(CV_CAP_PROP_POS_MSEC(10))。虽然这可能比这个问题更困难。 - GPPK

@GPPK，应该更像是cap.set（CV_CAP_PROP_POS_MSEC，123）frame = cap.read（）。如果像你提到的那样，set 能够正常工作，那么一切都没问题。 - Miki

Miki，我该如何将这种类型的图像直接转换为JPG格式，而无需保存？ - yusuf

Answer 1

你可以使用VideoCapture的set()函数。 你可以计算总帧数：

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

这里的7是prop-Id。你可以在这里找到更多信息http://docs.opencv.org/2.4/modules/highgui/doc/reading_and_writing_images_and_video.html

之后你就可以设置帧数，比如我想提取第100帧。

cap.set(cv2.CAP_PROP_FRAME_COUNT, 100)
ret, frame = cap.read()
cv2.imwrite("path_where_to_save_image", frame)

- Abhishek Sachan

---

很好！我想要说明一下，7是CV_CAP_PROP_FRAME_COUNT的序数值，而1是CV_CAP_PROP_POS_FRAMES的序数值 - 所以你在这里做的是将“帧阅读器”移动到第100帧的偏移量，然后读取“下一个”，也就是第101帧。 - SomethingSomething

有没有一种方法可以设置要提取的帧数？ - June Wang

如果从网络摄像头接收帧，会产生什么影响？ - Jacob Lawrence

魔法数字7实际上是常量“cv2.CAP_PROP_FRAME_COUNT”，即total_frames = cap.get(cv2.CAP_PROP_FRAME_COUNT)。 - Jabba

Answer 2

这是我的第一篇文章，如果我没有完全遵守协议，请不要严厉批评我。我只是想回复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)

- Eric De Luna

---

欢迎来到SO。解释你的代码如何回答问题总是一件好事。 - Nick

对我来说，这只是为range中的每个数字打印相同的图像。 - mikey