【python】多视频截取图片

多个视频根据时间间隔或者帧率间隔实现图片截取，每个视频保存一个文件夹

import os
import cv2

def cut_video():  # 截取视频中图片
    video_path = r'D:\Desktop\test\test\input/'  # 视频所在的路径
    f_save_path = r'D:\Desktop\test\test\output/'  # 保存图片的上级目录
    mask = 1  # 不同方法：1 按照帧率截取；2 按照时间截取(单位：s)；
    frame_interval = 40  # 设置帧率间隔
    time_interval = 1  # 设置时间间隔(单位：s)

    if (mask == 1):
        print("当前模式：按照帧率截取视频\n帧率间隔："+str(frame_interval))
    elif (mask == 2):
        print("当前模式：按照时间截取视频\n时间间隔："+str(time_interval)+"s")

    videos = os.listdir(video_path)
    for video_name in videos:
        file_name = video_name.split('.')[0]  # 拆分视频文件名称 ，剔除后缀
        folder_name = f_save_path + file_name  # 保存图片的上级目录+对应每条视频名称 构成新的目录存放每个视频
        os.makedirs(folder_name, exist_ok=True)  # 创建存放视频的对应目录
        vc = cv2.VideoCapture(video_path + video_name)  # 读入视频文件
        fps = vc.get(cv2.CAP_PROP_FPS)  # 获取当前视频帧率
        rval = vc.isOpened()  # 判断视频是否打开
        c = 1
        while rval:  # 循环读取视频帧
            rval, frame = vc.read()  # videoCapture.read() 函数，第一个返回值为是否成功获取视频帧，第二个返回值为返回的视频帧；
            pic_path = folder_name + '/'
            if rval:
                if(mask == 1):
                    if (c % round(frame_interval) == 0):  # 每隔frame_interval帧存储一次

                        cv2.imencode('.jpg', frame)[1].tofile(
                            pic_path + file_name + '_' + str(round(c / frame_interval)) + '.jpg')  # 中文路径也可以保存

                        print(file_name + '_' + str(round(c / frame_interval)) + '.jpg')
                    cv2.waitKey(1)
                    c = c + 1
                elif(mask == 2):
                    if (c % (round(fps)*time_interval) == 0):  # 每隔time_interval秒存储一次
                        cv2.imencode('.jpg', frame)[1].tofile(
                            pic_path + file_name + '_' + str(round(c / fps)) + '.jpg')
                        print(file_name + '_' + str(round(c / fps)) + '.jpg')
                    cv2.waitKey(1)
                    c = c + 1
            else:
                break
        vc.release()
        print('save_success' + folder_name)

cut_video()