opencv python 从视频中提取图片

墨蓝 2022-12-21 14:59 270阅读 0赞

此代码每30帧提取一张图片，如果你想设置成没帧提取一张则把frame\_interval设置为1，如要要设置为每秒提取一张，则把frame\_interval设置为你视频的帧率值。  
环境要求opencv python版本。

import cv2
    import logging
    
    # log information settings
    logging.basicConfig(level=logging.INFO,
                        format='%(asctime)s - %(levelname)s: %(message)s')
    
    
    def save_image(num, image):
        """Save the images. Args: num: serial number image: image resource Returns: None """
        image_path = 'E:/originsource/photo/{}.jpg'.format(str(num))
        cv2.imwrite(image_path, image)
    
    
    file_path = "E:/originsource/vedio/Capture00008-1280-720.avi"
    
    vc = cv2.VideoCapture(file_path)  # import video files
    
    # determine whether to open normally
    if vc.isOpened():
        # read 方法按帧读取视频
        # 其中 ret 是布尔值，表示是否正确读到帧。frame 就是每一帧的图像，是一个三维矩阵。
        ret, frame = vc.read()
    else:
        ret = False
    
    count = 0  # count the number of pictures
    frame_interval = 30  # video frame count interval frequency
    frame_interval_count = 0
    
    # loop read video frame
    while ret:
        ret, frame = vc.read()
        # store operation every time f frame
        if frame_interval_count % frame_interval == 0:
            save_image(count, frame)
            logging.info("num：" + str(count) + ", frame: " +
                         str(frame_interval_count))
            count += 1
        frame_interval_count += 1
        cv2.waitKey(1) #播放每一帧时适当持续时间
    
    vc.release()