当前位置：首页 > news >正文

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

news 来源：原创 2025/5/28 2:50:53

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")])