python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > Python中对FFmpeg封装开发库FFmpy

Python中对FFmpeg封装开发库FFmpy详解

作者:研创通之逍遥峰

这篇文章主要介绍了Python中对FFmpeg封装开发库FFmpy,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

FFmpy是一个轻量级的Python库,用于在Python脚本中方便地调用FFmpeg命令行工具。它提供了简洁的API接口来构建和执行FFmpeg命令,适合视频处理、音频转换等多媒体任务。下面将详细介绍FFmpy的功能和常见用法。

一、FFmpy简介与安装

1.1 FFmpy概述

FFmpy是一个简单的FFmpeg命令行包装器,它提供了一个Pythonic接口来编译FFmpeg命令行,并使用Python的subprocess模块来执行编译后的命令行。该项目的主要编程语言是Python,要求Python 3.8或更高版本。

1.2 安装方法

安装FFmpy非常简单,只需使用pip命令:

pip install ffmpy

建议在虚拟环境中安装FFmpy,以避免与其他项目的依赖冲突:

python -m venv myenv  # 创建虚拟环境
source myenv/bin/activate  # 激活虚拟环境(Linux/Mac)
myenv\Scripts\activate  # 激活虚拟环境(Windows)
pip install ffmpy

二、FFmpy核心类与方法

FFmpy中的FFmpeg类有两个核心方法:构造方法和run方法。

2.1 FFmpeg类构造方法

FFmpeg(executable='ffmpeg', global_options=None, inputs=None, outputs=None)

参数说明:

2.2 run方法

run(input_data=None, stdout=None, stderr=None)

参数说明:

三、FFmpy常见用法

3.1 视频格式转换

原始FFmpeg命令

ffmpeg -y -i input.webm -r 30 output.mp4

使用FFmpy实现

from ffmpy import FFmpeg

ff = FFmpeg(
    inputs={'input.webm': '-y'},
    outputs={'output.mp4': '-r 30'}
)
ff.run()

3.2 视频剪切

原始FFmpeg命令

ffmpeg -ss 15 -t 25 -i input.mp4 -c:v libx264 -c:a aac -strict experimental output.mp4

使用FFmpy实现

start = 15
end = 25
ff = FFmpeg(
    inputs={'input.mp4': f'-ss {start} -t {end}'},
    outputs={'output.mp4': '-c:v libx264 -c:a aac -strict experimental'}
)
ff.run()

也可以将剪切参数放在global_options中:

ff = FFmpeg(
    global_options={f'-ss {start} -t {end}'},
    inputs={'input.mp4': None},
    outputs={'output.mp4': '-c:v libx264 -c:a aac -strict experimental'}
)

3.3 多个视频合并

原始FFmpeg命令

ffmpeg -f concat -safe 0 -y -i filelist.txt -c copy -strict -2 output.mp4

使用FFmpy实现

ff = FFmpeg(
    inputs={'filelist.txt': '-f concat -safe 0 -y'},
    outputs={'output.mp4': '-c copy -strict -2'}
)
ff.run()

3.4 音频混合

原始FFmpeg命令

ffmpeg -y -i sound1.wav -i sound2.wav -filter_complex amix=inputs=2:duration=longest output.wav

使用FFmpy实现

ff = FFmpeg(
    inputs={'sound1.wav': '-y', 'sound2.wav': None},
    outputs={'output.wav': '-filter_complex amix=inputs=2:duration=longest'}
)
ff.run()

3.5 使用多线程提高性能

FFmpeg默认单线程执行,可以通过-threads参数使用多线程:

ff = FFmpeg(
    inputs={'input.mp4': '-y'},
    outputs={'output.mp4': '-c:v libx264 -threads 5 -preset ultrafast'}
)
ff.run()

四、高级用法与技巧

4.1 获取媒体文件元数据

使用FFprobe功能获取多媒体文件的元数据(尺寸、持续时间、比特率等):

import subprocess
import ffmpy
import json

source_path = "video.mp4"
tup_resp = ffmpy.FFprobe(
    inputs={source_path: None},
    global_options=[
        '-v', 'quiet',
        '-print_format', 'json',
        '-show_format', '-show_streams']
).run(stdout=subprocess.PIPE)

meta = json.loads(tup_resp[0].decode('utf-8'))
print(meta)

4.2 添加水印

ff = FFmpeg(
    inputs={'video.mp4': None, 'logo.png': None},
    outputs={'watermarked.mp4': '-filter_complex overlay=10:10'},
    global_options='-y'
)
ff.run()

4.3 从一系列图像创建视频

ff = FFmpeg(
    inputs={'image_%04d.jpg': '-framerate 24'},
    outputs={'output.mp4': '-c:v libx264 -pix_fmt yuv420p'}
)
ff.run()

4.4 使用FFmpy对USB相机进行截图的方法

FFmpy是FFmpeg的Python封装,可以方便地调用FFmpeg功能来处理USB相机视频流。以下是使用FFmpy从USB相机截图的详细方法:

4.4.1. 安装准备

首先确保已安装必要的库:

pip install ffmpy

同时需要安装FFmpeg(如果尚未安装):

# Ubuntu/Debian
sudo apt-get install ffmpeg

# macOS
brew install ffmpeg

# Windows
# 从 https://ffmpeg.org/download.html 下载并配置环境变量

4.4.2. 基础截图方法

from ffmpy import FFmpeg

# 获取USB相机设备(Linux通常为/dev/video0)
camera_device = "/dev/video0"

# 创建FFmpeg命令
ff = FFmpeg(
    inputs={camera_device: "-f v4l2 -framerate 30"},
    outputs={"snapshot.jpg": "-frames:v 1 -f image2"}
)

print("执行命令:", ff.cmd)
ff.run()  # 执行截图
ff = FFmpeg(
    inputs={camera_device: "-f v4l2 -framerate 30 -video_size 1280x720"},
    outputs={"highres_snapshot.jpg": "-frames:v 1 -f image2"}
)

4.4.3. 高级用法

import time
from ffmpy import FFmpeg

def capture_at_interval(interval=5, count=5):
    for i in range(count):
        timestamp = time.strftime("%Y%m%d_%H%M%S")
        ff = FFmpeg(
            inputs={camera_device: "-f v4l2"},
            outputs={f"snapshot_{timestamp}.jpg": "-frames:v 1 -f image2"}
        )
        ff.run()
        time.sleep(interval)

capture_at_interval(interval=2, count=3)  # 每2秒截图一次,共3张
from datetime import datetime

ff = FFmpeg(
    inputs={
        camera_device: "-f v4l2",
        # 添加时间戳滤镜
        None: "-f lavfi -i color=black:s=640x480 -filter_complex"
    },
    outputs={
        "timestamped.jpg": [
            "-filter_complex",
            "[0:v][1:v]overlay=10:10,drawtext=fontfile=/path/to/font.ttf:",
            "text='%(pts)':x=w-tw-10:y=h-th-10:fontcolor=white:fontsize=24",
            "-frames:v 1"
        ]
    }
)

4.4.4. 常见问题解决

设备无法打开

ff = FFmpeg(
    inputs={camera_device: "-f v4l2 -input_format mjpeg"},
    outputs={"quality_snapshot.jpg": "-q:v 2 -frames:v 1"}
)
# 第二个USB摄像头
ff = FFmpeg(
    inputs={"/dev/video1": "-f v4l2"},
    outputs={"second_cam.jpg": "-frames:v 1"}
)

4.4.5. 完整示例

from ffmpy import FFmpeg
import logging

def capture_usb_camera(
        device="/dev/video0",
        output="output.jpg",
        resolution="1280x720",
        framerate=30,
        quality=2
):
    """
    从USB相机捕获单帧图像
    
    参数:
        device: 相机设备路径
        output: 输出文件名
        resolution: 图像分辨率 (例如 "640x480")
        framerate: 帧率
        quality: 图像质量 (1-31, 1为最高质量)
    """
    try:
        ff = FFmpeg(
            inputs={
                device: f"-f v4l2 -video_size {resolution} -framerate {framerate}"
            },
            outputs={
                output: f"-frames:v 1 -q:v {quality} -f image2"
            }
        )
        logging.info(f"执行命令: {ff.cmd}")
        ff.run()
        logging.info(f"截图已保存到 {output}")
        return True
    except Exception as e:
        logging.error(f"截图失败: {str(e)}")
        return False

# 使用示例
if __name__ == "__main__":
    logging.basicConfig(level=logging.INFO)
    capture_usb_camera(
        device="/dev/video0",
        output="usb_snapshot.jpg",
        resolution="1920x1080",
        quality=1
    )

4.5 将USB相机转换为RTSP流

from ffmpy import FFmpeg
import logging
import signal
import sys

class CameraToRTSP:
    def __init__(self, device="/dev/video0", rtsp_url="rtsp://127.0.0.1:8554/live.sdp"):
        self.device = device
        self.rtsp_url = rtsp_url
        self.process = None
        
        # 设置信号处理
        signal.signal(signal.SIGINT, self.signal_handler)
        signal.signal(signal.SIGTERM, self.signal_handler)

    def start_stream(self):
        """启动RTSP流服务"""
        try:
            ff = FFmpeg(
                inputs={
                    self.device: "-f v4l2 -framerate 30 -video_size 1280x720 -input_format mjpeg"
                },
                outputs={
                    self.rtsp_url: [
                        "-c:v libx264 -preset ultrafast -tune zerolatency",
                        "-b:v 1500k -maxrate 2000k -bufsize 3000k",
                        "-f rtsp -rtsp_transport tcp"
                    ]
                }
            )
            
            logging.info(f"启动RTSP流: {self.rtsp_url}")
            logging.info(f"FFmpeg命令: {ff.cmd}")
            
            # 启动FFmpeg进程
            self.process = ff.run_async()
            
        except Exception as e:
            logging.error(f"启动RTSP流失败: {str(e)}")
            sys.exit(1)

    def signal_handler(self, signum, frame):
        """处理终止信号"""
        logging.info("接收到终止信号,停止RTSP流...")
        if self.process:
            self.process.terminate()
        sys.exit(0)

if __name__ == "__main__":
    logging.basicConfig(
        level=logging.INFO,
        format='%(asctime)s - %(levelname)s - %(message)s'
    )
    
    # 创建并启动RTSP流
    streamer = CameraToRTSP(
        device="/dev/video0",
        rtsp_url="rtsp://127.0.0.1:8554/live.sdp"
    )
    streamer.start_stream()

五、常见问题解决

5.1 安装问题

5.2 命令行编译错误

5.3 运行时错误

六、FFmpy与其他库的配合

虽然FFmpy本身就是一个生态中的组件,但它可以与其他Python多媒体处理库配合使用,如moviepy、OpenCV等,用于更复杂的媒体处理场景。例如:

七、总结

FFmpy作为FFmpeg的Python包装器,提供了简洁的API来执行各种多媒体处理任务。它的主要优势包括:

  1. Pythonic接口:比直接使用FFmpeg命令行更符合Python开发习惯
  2. 灵活性:可以轻松构建复杂的FFmpeg命令
  3. 集成性:便于将FFmpeg功能集成到Python项目中

通过本文的介绍,您应该已经掌握了FFmpy的核心功能和常见用法。在实际项目中,可以根据具体需求选择合适的参数和选项,充分发挥FFmpeg的强大功能。

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

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