视频文件夹下批量抽帧

m0_60790028

已于 2023-12-12 16:42:49 修改

阅读量429

点赞数 10

文章标签：音视频 opencv 计算机视觉

于 2023-12-12 16:29:30 首次发布

本文链接：https://blog.csdn.net/m0_60790028/article/details/134952901

版权

文章描述了一个使用Python的VideoFrameExtractor类，通过OpenCV库从MP4视频中每`timeF`帧提取图片并保存到指定文件夹的过程。代码示例展示了如何在给定的视频文件夹中执行这个帧提取操作。

摘要由CSDN通过智能技术生成

import cv2
import os
import glob


class VideoFrameExtractor:
    def __init__(self):
        self.start_name = '0000000.jpg'
        self.timeF = 10

    def extract_frames(self, video_path, save_path):
        video_files = glob.glob(os.path.join(video_path, '*.mp4'))  # 获取目标文件夹下所有的MP4文件路径

        for video_file in video_files:
            cv = cv2.VideoCapture(video_file)  # 读入视频文件，命名为 cv

            if cv.isOpened():  # 判断是否正常打开
                rval, frame = cv.read()
                i = 1
                n = 0  # 重置帧计数器
            else:
                rval = False
                print(f'无法打开视频文件: {video_file}')
                continue

            video_name = os.path.splitext(os.path.basename(video_file))[0]  # 获取视频文件名（不带后缀）

            while rval:  # 正常打开 开始处理
                rval, frame = cv.read()
                jpg_name = os.path.join(save_path, f"{video_name}_{str(n).zfill(6)}.jpg")  # 命名保存的图片

                if (i % self.timeF == 0):  # 每隔timeF帧进行存储操作
                    try:
                        cv2.imwrite(jpg_name, frame)  # 存储为图像
                    except Exception as e:
                        print(f'保存图片出错: {e}')
                    n += 1
                i += 1
            cv2.waitKey(1)
            cv.release()
            print(f'{video_file} 处理完成')
        print('抽帧完成')


if __name__ == "__main__":
    video_folder_path = 'E:/tst1212'  # 设置为视频文件存储的目标文件夹路径
    img_output_path = 'E:/piliang'

    if not os.path.exists(img_output_path):  # 如果存储图片的文件夹不存在，自动创建保存图片文件夹
        os.makedirs(img_output_path)

    frame_extractor = VideoFrameExtractor()
    frame_extractor.extract_frames(video_folder_path, img_output_path)  # 执行提取图片程序