python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > python ffmpeg 关键帧抽取

基于Linux的ffmpeg python的关键帧抽取

作者：lambdarw

本文主要介绍了基于Linux的ffmpeg python的关键帧抽取,实现以按帧或时间间隔抽取关键帧,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

1.FFmpeg的环境配置

首先强调，ffmpeg-python包与ffmpeg包不一样。

1) 创建一个虚拟环境env

conda create -n yourenv python=3.x
conda activate yourenv

2) ffmpeg-python包的安装

pip install ffmpeg-python

3) 安装系统级别的 FFmpeg 工具

虽然安装了 ffmpeg-python 包（也就是 Python 的 wrapper），但是系统里没有安装实际的 FFmpeg 可执行程序，会导致程序无法执行 ffmpeg 工具。

🧪在系统中安装 FFmpeg 二进制文件

cd ./your_env_path
conda install -c conda-forge ffmpeg

🔁安装完成后，验证 FFmpeg 是否可用

ffmpeg -version

4) FFmpeg 添加到系统变量

✅首先，在终端确认 FFmpeg 安装位置

which ffmpeg

你应该会看到类似的输出结果：

/usr/bin/ffmpeg
# 如果是在 Conda 环境下安装的，则输出：
/your_env_path/envs/yourenv/bin/ffmpeg  
# 例如：/home/zrw/anaconda3/envs/condetect/bin/ffmpeg

✅第二步，临时验证 PATH 设置在 Python 脚本中 运行前添加如下代码，确保 FFmpeg 所在路径加入环境变量：

import os

# 添加 FFmpeg 路径（改成你实际的 ffmpeg 路径）
os.environ["PATH"] += os.pathsep + "/your_env_path/envs/yourenv/bin/ffmpeg"
# 例如：os.environ["PATH"] += os.pathsep + "/home/zrw/anaconda3/envs/condetect/bin/ffmpeg"

放在程序中调用 ffmpeg之前就可以了。

✅ 第三步（可选）：永久添加到 PATH

export PATH=$PATH:/yourenv_path/envs/yourenv/bin

然后运行：

source ~/.bashrc

5) 验证安装是否成功

建立一个python文件，输入以下代码，正确输出即可。

import ffmpeg
print(ffmpeg.__file__)  # 应该输出路径，比如 .../site-packages/ffmpeg/__init__.py
print(dir(ffmpeg))      # 应该能看到 ['input', 'output', 'run', ...]

打印结果如下：

在这里插入图片描述

2.主程序

🚀引入环境

import os
import ffmpeg
import numpy as np
from PIL import Image

# 添加 FFmpeg 路径（改成你实际的 ffmpeg 路径）
os.environ["PATH"] += os.pathsep + "/home/zrw/anaconda3/envs/condetect/bin"

✅按固定帧数间隔抽帧

def extract_keyframes(video_path, output_dir, frame_interval=120):
os.makedirs(output_dir, exist_ok=True)
 (
      ffmpeg.input(video_path)
      .filter("select", f"not(mod(n,{frame_interval}))")  # 每N帧取1帧
      .output(os.path.join(output_dir, "frame_%04d.png"))
      .run(quiet=True)
)
 return sorted([f for f in os.listdir(output_dir) if f.endswith(".png")])

✅按固定时间间隔（秒）抽帧

def extract_keyframes(video_path, output_dir, interval_seconds=8):
    os.makedirs(output_dir, exist_ok=True)
    (
        ffmpeg.input(video_path)
        .filter("fps", fps=1/interval_seconds)  # 如 interval_seconds=8 表示每8秒1帧
        .output(os.path.join(output_dir, "frame_%04d.png"))
        .run(quiet=True)
    )
    return sorted([f for f in os.listdir(output_dir) if f.endswith(".png")])

到此这篇关于基于Linux的ffmpeg python的关键帧抽取的文章就介绍到这了,更多相关python ffmpeg 关键帧抽取内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！ 

您可能感兴趣的文章:
阅读全文