OpenCV - захватить произвольный кадр из видеофайла

Question

Я использую следующий код для извлечения конкретного кадра из видеофайла. В этом примере я просто получаю средний кадр:

import cv2
video_path = '/tmp/wonderwall.mp4'
vidcap = cv2.VideoCapture(video_path)
middle_frame = int(vidcap.get(cv2.CAP_PROP_FRAME_COUNT) / 2)
success, image = vidcap.read()
count = 0
success = True
while success:
    success, image = vidcap.read()
    if count == middle_frame:
        temp_file = tempfile.NamedTemporaryFile(suffix='.jpg', delete=False)
        cv2.imwrite(temp_file.name, image)
    count += 1

Однако при использовании этого метода захват среднего кадра в очень большом файле может занять некоторое время.

Видимо, в старшем cv модуль, можно сделать:

import cv
img = cv.QueryFrame(capture)

Есть ли подобный способ в cv2 захватить определенный кадр в видеофайле, без необходимости перебирать все кадры?

Answer 1

Вы можете сделать это таким же образом, в C++ (преобразование Python должно быть более легким).

cv::VideoCapture cap("file.avi");
double number_of_frame = cap.get(CV_CAP_PROP_FRAME_COUNT);
cap.set(CV_CAP_PROP_POS_FRAMES, IndexOfTheFrameYouWant);
cv::Mat frameIwant = cap.read();

Для справки:

VideoCapture::get(int propId)

Может принимать различные флаги, возвращая почти все, что вы можете пожелать ( http://docs.opencv.org/2.4/modules/highgui/doc/reading_and_writing_images_and_video.html и искать get ()).

 VideoCapture::set(int propId, double value)

Set будет делать то, что вы хотите (то же самое, что и поиск документов для set), если вы используете propID 1 и индекс желаемого фрейма. Вы должны заметить, что если индекс, который вы используете в качестве параметра, превосходит максимальный кадр, то код захватит последний кадр видео, если вам повезет или произойдет сбой во время выполнения.