Python脚本020：将视频按帧提取图片

Input:将/home/xpy/xl/workstation/Car/Dataset/Dataset_Realworld/目录下的.mp4文件

Output:切成一帧一帧的图片放到同名的文件下

import cv2
import os
from IPython import embed

def save_img():
    video_path = r'/home/xpy/xl/workstation/Car/Dataset/Dataset_Realworld/'
    videos = os.listdir(video_path)
    # embed()
    for video_name in videos:
        file_name = video_name.split('.')[0]
        folder_name = video_path + file_name
        os.makedirs(folder_name,exist_ok=True)
        vc = cv2.VideoCapture(video_path+video_name) #读入视频文件
        c=0
        rval=vc.isOpened()

        while rval:   #循环读取视频帧
            c = c + 1
            rval, frame = vc.read()
            pic_path = folder_name+'/'
            if rval:
                cv2.imwrite(pic_path + file_name + '_' + str(c) + '.jpg', frame) #存储为图像,保存名为 文件夹名_数字（第几个文件）.jpg
                cv2.waitKey(1)
            else:
                break
        vc.release()
        print('save_success')
        print(folder_name)


if __name__ == '__main__':
    save_img()