Python批量将视频帧提取为图片的实现方法

向北走

发布时间：2026-10-15| 4人浏览过

原创

本文介绍使用Python和OpenCV库将视频文件批量转换为图片序列的方法。通过配置源视频路径，脚本自动遍历文件夹，按视频名称创建子目录，并将视频帧保存为JPG格式。文中提供了完整的Python代码示例、文件结构说明及终端运行日志解读，帮助开发者快速实现视频内容到静态图片的转换。

前言

代码虽然有时让人头疼，但我逐渐爱上了其严谨的逻辑以及解决实际问题时的力量。（注：本专题配图经过缩放处理后，与我的头像有些相似之处。）

代码展示

以下代码实现了从多个文件夹中提取多个视频片段并转换为图片的功能，具体逻辑可参考代码注释。操作步骤如下：将代码复制到新建的 main.py 文件中，确保第63行配置的源文件路径正确无误，随后运行脚本即可。

# 将四个文件夹中的多个视频文件转换为图片
import cv2
import os


number_in_dir = []  # 可以去掉
def video2image(video_path, image_save_dir):
    # 1. 将图片的存储路径加上源文件的文件夹名，如'Coffee_room_01'等。
    if os.path.exists(video_path):  # 判断源路径是否正确
        print(video_path + '\t ok')
        if video_path.split('\\')[-2] != 'Videos':  # 这个根据我文件路径自行添加的
            image_save_dir += video_path.split('\\')[-2] + '\\'
        else:
            image_save_dir += video_path.split('\\')[-3] + '\\'

        number_in_dir.append(len(os.listdir(video_path)))  # 可以去掉
    else:
        print(video_path + ' \033[0;37;41merror\033[0m')
        return

    # 2. 依次读取源文件里的文件，如果后缀名是‘avi'或 'MP3'，则创建一个关于文件名的子文件夹
    files_in_video_path_list = os.listdir(video_path)
    print(len(files_in_video_path_list))  # 获取当前文件夹下文件个数，用来判断是否有其他类型的文件
    file_count = 0  # 用于统计个数，验证是否全为视频文件，会与len(files_in_video_path_list)进行比较

    for file_in_video_path in files_in_video_path_list:
        file_name = os.path.basename(file_in_video_path)
        if file_name.split('.')[-1] == 'avi' or file_name.split('.')[-1] == 'mp3':
            file_count += 1  # 视频文件数+1
            every_video_save_dir = file_name.split('.')[0] + '\\'
            if not os.path.exists(image_save_dir + every_video_save_dir):  # 创建属于相应文件夹的存储路径
                os.makedirs(image_save_dir + every_video_save_dir)
        else:
            print(' \033[0;37;41merror\033[0m')
            break

        # 3. 打印正在处理文件的序号和他的文件名，并开始转换
        print(str(file_count) + ':' + file_name)
        cap = cv2.VideoCapture(video_path + file_name)

        flag = cap.isOpened()
        if not flag:
            print("open" + video_path + file_name + "error!")

        frame_count = 0  # 给每一帧标号
        while True:
            frame_count += 1
            flag, frame = cap.read()
            if not flag:  # 如果已经读取到最后一帧则退出
                break
            if os.path.exists(image_save_dir + every_video_save_dir + str(frame_count) + '.jpg'):  # 在源视频不变的情况下，如果已经创建，则跳过
                break
            cv2.imwrite(image_save_dir + every_video_save_dir + str(frame_count) + '.jpg', frame)

        cap.release()
        print(file_name + ' save to ' + image_save_dir + every_video_save_dir + 'finished ')  # 表示一个视频片段已经转换完成
    if file_count != len(files_in_video_path_list):
        print(' \033[0;37;41Maybe loss something\033[0m')


if __name__ == '__main__':
    # 需要转换的文件路径列表(自定义修改）
    video_path_list = ['E:\\Datasets_FallDetection\\2 LE2I\Coffee_room_01\\Coffee_room_01\\Videos\\',
                       'E:\\Datasets_FallDetection\\2 LE2I\Home_01\\Home_01\Videos\\',
                       'E:\\Datasets_FallDetection\\2 LE2I\Lecture_room\\Lecture room\\',
                       'E:\\Datasets_FallDetection\\2 LE2I\\Office\\Office\\']

    # 预期存储在的主文件夹，即'result'文件夹
    image_save_dir = '.\\result\\'

    # 进行转换
    for video_path in video_path_list:
        video2image(video_path, image_save_dir)