Python 视频抽帧处理

根据需求对视频进行抽帧处理,将抽出来的图片保存到指定位置,代码如下

#引用的依赖
import cv2
import os


def extract_frames(video_path, output_folder, interval_seconds=1, fps_video=None):
    """
    从视频文件中抽帧
    :param video_path: 视频文件的路径
    :param output_folder: 存储帧的文件夹路径
    :param interval_seconds: 每多少秒抽取一帧
    :param fps_video: 视频的帧率（可选，如果未提供，则会自动从视频中读取）
    """
    # 确保输出文件夹存在
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

        # 使用OpenCV读取视频
    cap = cv2.VideoCapture(video_path)

    # 如果未提供视频的帧率，则从视频中读取
    if fps_video is None:
        fps_video = int(cap.get(cv2.CAP_PROP_FPS))

        # 计算两个帧之间的间隔（基于时间间隔）
    frame_interval = max(1, int(fps_video * interval_seconds))

    # 当前帧的索引
    frame_idx = 0

    # 遍历视频帧
    while True:
        ret, frame = cap.read()
        if not ret:
            break

            # 如果当前帧的索引是间隔的倍数，则保存该帧
        if frame_idx % frame_interval == 0:
            filename = os.path.basename(video_path)  # 获取文件名和扩展名
            filename_without_ext = os.path.splitext(filename)[0]  # 去掉扩展名
            # 构造帧的文件名（例如：frame1.jpg, frame2.jpg...）
            filename = os.path.join(output_folder, f'{filename_without_ext}_{frame_idx // frame_interval}.jpg')
            cv2.imwrite(filename, frame)
            print(f'Saved {filename}')

        frame_idx += 1

        # 释放视频文件
    cap.release()


if __name__ == '__main__':
    # 使用示例
    video_path = '文件路径/123.mp4'
    output_folder = '生成图片的路径'
    interval_seconds = 1  # 每5秒抽一帧
    fps_video = 1  # 视频的帧率（可选，如果未提供，则会自动从视频中读取）
    extract_frames(video_path, output_folder, interval_seconds, fps_video)