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

            # 如果当前帧的索引是间隔的倍数，则保存该帧