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

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

pip install ffmpeg-python

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

ffmpeg -version

which ffmpeg

/usr/bin/ffmpeg
# 如果是在 Conda 环境下安装的，则输出：
/your_env_path/envs/yourenv/bin/ffmpeg  
# 例如：/home/zrw/anaconda3/envs/condetect/bin/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"

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

source ~/.bashrc

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

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