【python】多视频截取图片

知五年三

已于 2023-03-03 23:34:28 修改

阅读量1.2k

点赞数 3

文章标签： python 音视频开发语言

于 2023-03-03 23:30:46 首次发布

本文链接：https://blog.youkuaiyun.com/qq_42506719/article/details/129329010

版权

该代码段使用Python的OpenCV库实现从多个视频中按照设定的帧率或时间间隔截取图片，每个视频的截图保存在对应的文件夹中。用户可以选择按照帧率间隔（例如每40帧）或时间间隔（例如每1秒）进行截图。截取的图片以.jpg格式存储，并按顺序命名。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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()