opencv读取本地视频， 并保存视频帧

C++

#include <opencv2/highgui/highgui.hpp>
#include <opencv2/imgcodecs/imgcodecs.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>

using namespace std;
using namespace cv;
int main() {
    const char save_dir[256] = "pics/";
    if (access(save_dir, 0) == -1) {
        cout << "目录不存在, 创建目录" << endl;
        mkdir(save_dir, S_IRUSR | S_IWUSR | S_IXUSR | S_IRWXG | S_IRWXO);
    }
    VideoCapture cap;
    Mat frame;
    frame = cap.open("../output.mp4");
    if(!cap.isOpened()){
        printf("cat open \n");
        return -1;
    }
    int frame_count = 0;
    int save_count = 0;
    while (cap.read(frame) and save_count < 10){
        frame_count += 1;
        if(frame_count % 30 == 0){
            string save_name = save_dir + to_string(save_count) + ".jpg";
            imwrite(save_name, frame);
            save_count += 1;
        }
    }
    return 0;
}

Python

import os.path as osp
import os
import cv2

if __name__ == '__main__':

    save_dir = 'pics'
    if not osp.exists(save_dir):
        print('目录不存在，　创建目录')
        os.mkdir(save_dir)
    video_cap = cv2.VideoCapture('../output.mp4')
    fps = video_cap.get(cv2.CAP_PROP_FPS)
    width = video_cap.get(cv2.CAP_PROP_FRAME_WIDTH)  # video width
    height = video_cap.get(cv2.CAP_PROP_FRAME_HEIGHT)  # video height
    frame_number = video_cap.get(7)  # number of frames
    duration = int(frame_number // fps)
    
    frame_count = 0
    save_count = 0
    while True and save_count < 10:
        frame_count += 1
        ret, frame = video_cap.read()
        if ret is False:
            break
        if frame_count % 30 == 0:
            save_name = osp.join(save_dir, str(save_count) + '.jpg')
            cv2.imwrite(save_name, frame)
            save_count += 1