【干货】视频文件抽帧(opencv和ffmpeg方式对比)

1 废话不多说,直接上代码

opencv方式

python 复制代码
import time
import subprocess
import cv2, os
from math import ceil

def extract_frames_opencv(video_path, output_folder, frame_rate=1):
    """
    使用 OpenCV 从视频中抽取每秒指定帧数的帧,并保存到指定文件夹。
    如果视频长度不是整数秒,则会在最后一帧时补充空白图像。

    参数:
    video_path (str): 输入视频文件的路径。
    output_folder (str): 输出帧图像文件的文件夹路径。
    frame_rate (int): 每秒抽取的帧数,默认为 1。

    返回:
    None
    """
    start_time = time.time()
    # 创建输出文件夹
    os.makedirs(output_folder, exist_ok=True)

    # 打开视频文件
    cap = cv2.VideoCapture(video_path)

    # 获取视频长度和帧率
    fps = cap.get(cv2.CAP_PROP_FPS)
    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
    duration = total_frames / fps

    # 计算需要抽取的总帧数
    target_frames = int(duration * frame_rate)

    # 逐帧抽取图像
    frame_idx = 0
    for i in range(target_frames):
        cap.set(cv2.CAP_PROP_POS_FRAMES, int(i * fps / frame_rate))
        ret, frame = cap.read()
        if ret:
            cv2.imwrite(os.path.join(output_folder, f"frame_{frame_idx:06d}.jpg"), frame)
            frame_idx += 1
        else:
            break

    # 如果最后一帧不是完整的一帧,则补充空白图像
    if frame_idx < target_frames:
        for i in range(frame_idx, target_frames):
            blank_image = 255 * np.ones((int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)), int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), 3), dtype=np.uint8)
            cv2.imwrite(os.path.join(output_folder, f"frame_{i:06d}.jpg"), blank_image)

    # 释放视频捕获对象
    cap.release()

    print(f"成功从视频中抽取了 {target_frames} 帧, 一共耗时{time.time() - start_time}s")

ffmpeg方式

python 复制代码
def extract_frames_ffmpeg(video_path, output_folder, frame_rate=1):
    """
    使用 FFmpeg 从视频中抽取每秒指定帧数的帧,并保存到指定文件夹。
    如果视频长度不是整数秒,则会抛出异常。

    参数:
    video_path (str): 输入视频文件的路径。
    output_folder (str): 输出帧图像文件的文件夹路径。
    frame_rate (int): 每秒抽取的帧数,默认为 1。

    返回:
    None
    """
    start_time = time.time()
    # 创建输出文件夹
    os.makedirs(output_folder, exist_ok=True)

    # 获取视频长度
    command = ["ffprobe", "-v", "error", "-show_entries", "format=duration", "-of",
               "default=nokey=1:noprint_wrappers=1", video_path]
    result = subprocess.run(command, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    if result.returncode != 0:
        raise ValueError("Failed to get video duration.")
    duration = float(result.stdout.decode().strip())

    # 四舍五入视频长度到最接近的整数秒
    duration = round(duration)

    # 构建 FFmpeg 命令
    command = [
        "ffmpeg",
        "-i", video_path,
        "-vf", f"fps={frame_rate}",
        "-frames:v", "%d" % (ceil(duration * frame_rate)),
        os.path.join(output_folder, "frame_%06d.jpg")
    ]

    # 执行 FFmpeg 命令
    subprocess.run(command, check=True)

    print(f"成功从视频中抽取了 {ceil(duration * frame_rate)} 帧, 一共耗时{time.time() - start_time}s")

2 测试实验对比

测试一个56s的mp4文件

复制代码
extract_frames_opencv(video_path, output_folder, 1)
# 成功从视频中抽取了 55 帧, 一共耗时10.131151914596558s
extract_frames_ffmpeg(video_path, output_folder, 1)
# 成功从视频中抽取了 56 帧, 一共耗时8.075150966644287s

1帧/s时,ffmpeg稍快2s

复制代码
extract_frames_opencv(video_path, output_folder1, 5)
# 成功从视频中抽取了 278 帧, 一共耗时54.822526931762695s
extract_frames_ffmpeg(video_path, output_folder2, 5)
# 成功从视频中抽取了 280 帧, 一共耗时8.546468019485474s

5帧/s时,ffmpeg方式只增加0.5s,opencv增加了5倍时长

在大文件抽帧、或者抽帧频率较高时,ffmpeg效率更高。

3 ffmpeg抽帧图片更小?

对比了下抽帧图片,分辨率一致,但是ffmpeg抽帧图片好像小很多,为什么呢?

搜了下官方解释:

FFmpeg 和 OpenCV 在抽取视频帧并保存图像时,可能会产生不同的结果,包括文件大小和图像质量等。出现这种差异的原因主要有以下几点:

  1. 编码器选择 :

    • FFmpeg 默认使用 JPEG 编码器,这种编码方式通常可以生成较小的文件大小,但可能会略有些图像质量损失。
    • OpenCV 默认使用无损的 PNG 编码器,生成的文件通常会更大,但图像质量更好。
  2. 色彩空间转换 :

    • FFmpeg 在抽取帧时,可能会将视频的色彩空间转换为更适合 JPEG 编码的格式,例如 YUV 420。这种转换可以进一步减小文件大小。
    • OpenCV 则可能会保留视频原有的色彩空间,如 RGB,在保存为 PNG 时不进行额外的转换。
  3. 图像缩放和重采样 :

    • FFmpeg 在抽取帧时,可能会根据输出分辨率对图像进行缩放和重采样,从而进一步优化文件大小。
    • OpenCV 则可能会直接保存原始分辨率的图像,不进行任何缩放处理。

综上所述,FFmpeg 在抽取视频帧并保存为图像时,通常会采取一些优化措施,如使用 JPEG 编码、色彩空间转换和图像缩放等,从而生成相对较小的文件大小。而 OpenCV 则更倾向于保留原始的视觉质量,因此生成的图像文件会相对更大。

相关推荐
qq_416276421 分钟前
LOFAR物理频谱特征提取及实现
人工智能
余俊晖31 分钟前
如何构造一个文档解析的多模态大模型?MinerU2.5架构、数据、训练方法
人工智能·文档解析
Akamai中国2 小时前
Linebreak赋能实时化企业转型:专业系统集成商携手Akamai以实时智能革新企业运营
人工智能·云计算·云服务
LiJieNiub3 小时前
读懂目标检测:从基础概念到主流算法
人工智能·计算机视觉·目标跟踪
weixin_519535773 小时前
从ChatGPT到新质生产力:一份数据驱动的AI研究方向指南
人工智能·深度学习·机器学习·ai·chatgpt·数据分析·aigc
爱喝白开水a4 小时前
LangChain 基础系列之 Prompt 工程详解:从设计原理到实战模板_langchain prompt
开发语言·数据库·人工智能·python·langchain·prompt·知识图谱
takashi_void4 小时前
如何在本地部署大语言模型(Windows,Mac,Linux)三系统教程
linux·人工智能·windows·macos·语言模型·nlp
OpenCSG4 小时前
【活动预告】2025斗拱开发者大会,共探支付与AI未来
人工智能·ai·开源·大模型·支付安全
生命是有光的4 小时前
【深度学习】神经网络基础
人工智能·深度学习·神经网络
数字供应链安全产品选型4 小时前
国家级!悬镜安全入选两项“网络安全国家标准应用实践案例”
人工智能·安全·web安全