Python结合ffmpeg 实现单线程和多线程推流

更新时间：2025年06月17日 10:21:19 作者：浪浪山小白兔

本文主要介绍了Python结合ffmpeg 实现单线程和多线程推流,将通过两个不同的实现方式,即单线程推流和多线程推流,来展示如何利用 cv2（OpenCV）和 subprocess 等库将视频帧推送到指定的 RTMP 地址,感兴趣的可以了解一下

一、引言

在本文中，我们将详细介绍如何使用 Python 进行视频的推流操作。我们将通过两个不同的实现方式，即单线程推流和多线程推流，来展示如何利用 cv2（OpenCV）和 subprocess 等库将视频帧推送到指定的 RTMP 地址。这两种方式都涉及到从摄像头读取视频帧，以及使用 ffmpeg 命令行工具将视频帧进行编码和推流的过程。

二、单线程推流

以下是单线程推流的代码：

import cv2 as cv
import subprocess as sp


def push_stream():
    # 视频读取对象
    cap = cv.VideoCapture(0) 
    fps = int(cap.get(cv.CAP_PROP_FPS))
    w = int(cap.get(cv.CAP_PROP_FRAME_WIDTH))
    h = int(cap.get(cv.CAP_PROP_FRAME_HEIGHT))
    ret, frame = cap.read()
    # 推流地址
    rtmpUrl = "rtmp://192.168.3.33:1935/live/"
    # 推流参数
    command = ['ffmpeg',
              '-y',
              '-f', 'rawvideo',
              '-vcodec','rawvideo',
              '-pix_fmt', 'bgr24',
              '-s', "{}x{}".format(w, h),
              '-r', str(fps),
              '-i', '-',
              '-c:v', 'libx264',
              '-pix_fmt', 'yuv420p',
              '-preset', 'ultrafast',
              '-f', 'flv', 
              rtmpUrl]
    # 创建、管理子进程
    pipe = sp.Popen(command, stdin=sp.PIPE, bufsize=10 ** 8)
    # 循环读取
    while cap.isOpened():
        # 读取一帧
        ret, frame = cap.read()
        if frame is None:
            print('read frame err!')
            continue
        # 显示一帧
        cv.imshow("frame", frame)
        # 按键退出
        if cv.waitKey(1) & 0xFF == ord('q'):
            break
        # 读取尺寸、推流
        # img=cv.resize(frame,size)
        pipe.stdin.write(frame) 
    # 关闭窗口
    cv.destroyAllWindows()
    # 停止读取
    cap.release()

在这个单线程的实现中，我们执行以下步骤：

初始化视频读取对象：

使用 cv2.VideoCapture(0) 来打开默认的摄像头设备。
获取摄像头的帧率 fps、宽度 w 和高度 h 等参数。

设置推流地址和参数：

定义 rtmpUrl 作为推流的目标地址。
构造 ffmpeg 的命令列表 command，该列表包含了一系列的参数，如 -y 表示覆盖输出文件、-f rawvideo 表示输入格式为原始视频等。
使用 sp.Popen 创建一个子进程，将 ffmpeg 命令作为子进程运行，并且将其输入管道 stdin 连接到我们的程序。

循环读取和推流：

在一个 while 循环中，不断读取摄像头的帧。
若读取失败，打印错误信息并继续。
使用 cv2.imshow 显示当前帧，同时监听 q 键，按下 q 键时退出程序。
将读取到的帧通过管道发送给 ffmpeg 进行推流。

三、多线程推流

以下是多线程推流的代码：

import queue
import threading
import cv2 as cv
import subprocess as sp


class Live(object):
    def __init__(self):
        self.frame_queue = queue.Queue()
        self.command = ""
        # 自行设置
        self.rtmpUrl = ""
        self.camera_path = ""

    def read_frame(self):
        print("开启推流")
        cap = cv.VideoCapture(self.camera_path)

        # Get video information
        fps = int(cap.get(cv.CAP_PROP_FPS))
        width = int(cap.get(cv.CAP_PROP_FRAME_WIDTH))
        height = int(cap.get(cv.CAP_PROP_FRAME_HEIGHT))

        # ffmpeg command
        self.command = ['ffmpeg',
                       '-y',
                       '-f', 'rawvideo',
                       '-vcodec','rawvideo',
                       '-pix_fmt', 'bgr24',
                       '-s', "{}x{}".format(width, height),
                       '-r', str(fps),
                       '-i', '-',
                       '-c:v', 'libx264',
                       '-pix_fmt', 'yuv420p',
                       '-preset', 'ultrafast',
                       '-f', 'flv', 
                       self.rtmpUrl]

        # read webcamera
        while(cap.isOpened()):
            ret, frame = cap.read()
            if not ret:
                print("Opening camera is failed")
                break

            # put frame into queue
            self.frame_queue.put(frame)

    def push_frame(self):
        # 防止多线程时 command 未被设置
        while True:
            if len(self.command) > 0:
                # 管道配置
                p = sp.Popen(self.command, stdin=sp.PIPE)
                break

        while True:
            if self.frame_queue.empty()!= True:
                frame = self.frame_queue.get()
                # process frame
                # 你处理图片的代码
                # write to pipe
                p.stdin.write(frame.tostring())

    def run(self):
        threads = [
            threading.Thread(target=Live.read_frame, args=(self,)),
            threading.Thread(target=Live.push_frame, args=(self,))
        ]
        [thread.setDaemon(True) for thread in threads]
        [thread.start() for thread in threads]

在这个多线程的实现中，我们使用了 threading 和 queue 库：

类的初始化：

创建一个 Live 类，在 __init__ 方法中初始化帧队列 frame_queue、command、rtmpUrl 和 camera_path 等变量。

读取帧的线程方法：

read_frame 方法中，使用 cv2.VideoCapture(self.camera_path) 打开摄像头。
获取摄像头的参数，并构造 ffmpeg 命令。
不断从摄像头读取帧，并将帧放入队列 frame_queue 中。

推流的线程方法：

push_frame 方法中，等待 command 被设置，然后使用 sp.Popen 启动 ffmpeg 子进程。
从帧队列中取出帧，并将其写入 ffmpeg 的输入管道进行推流。

启动线程：

run 方法创建并启动两个线程，一个用于读取帧，一个用于推流，并且将它们设置为守护线程。

四、代码解释和注意事项

单线程推流：

这种方式相对简单，适合初学者理解。但由于是单线程操作，在处理复杂任务时可能会导致性能瓶颈，特别是在同时进行视频显示、读取和推流的情况下，可能会出现卡顿现象。

多线程推流：

利用多线程可以将不同的任务分配给不同的线程，提高性能。
frame_queue 是一个线程安全的队列，用于在两个线程之间传递帧数据，避免了数据竞争问题。
setDaemon(True) 使得线程在主线程结束时自动终止，防止程序无法正常退出。

五、总结

通过上述代码和解释，我们可以看到如何使用 Python 进行单线程和多线程的视频推流操作。单线程代码简单明了，但性能可能受限；多线程代码可以更好地处理高负载，但也需要注意线程安全和资源管理等问题。在实际应用中，我们可以根据具体的需求和硬件性能来选择合适的推流方式。同时，我们可以进一步优化代码，例如添加异常处理、优化帧处理逻辑等，以提高程序的稳定性和性能。

到此这篇关于Python结合ffmpeg 实现单线程和多线程推流的文章就介绍到这了,更多相关Python ffmpeg 单线程和多线程推流内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:

Python的闭包和装饰器你真的了解吗
这篇文章主要为大家详细介绍了Python的闭包和装饰器，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下，希望能够给你带来帮助
2022-02-02
零基础也能轻松上手的Python自动化办公实战教学(新手必看)
这篇文章为Python零基础用户提供了3个实用的自动化办公项目,帮助解决重复性工作问题,特别适合被重复性办公工作困扰的职场人士和想提升效率的零基础用户,希望对大家有所帮助
2026-06-06
Python键盘输入转换为列表的实例
今天小编就为大家分享一篇Python键盘输入转换为列表的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2018-06-06
Python爬虫使用浏览器cookies：browsercookie过程解析
这篇文章主要介绍了Python爬虫使用浏览器cookies：browsercookie,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-10-10
pytorch中的dataset用法详解
这篇文章主要介绍了pytorch的dataset用法详解,当我们继承了一个 Dataset类之后，我们需要重写 len 方法，该方法提供了dataset的大小； getitem 方法，该方法支持从 0 到 len(self)的索引,下面来看看附有代码的讲解吧，希望对你的学习或者工作有所帮助
2022-01-01
python数据挖掘需要学的内容
在本篇文章中我们给大家整理了关于python数据挖掘需要学什么的知识点指南，有兴趣的朋友们跟着参考下。
2019-06-06
Python实现遍历目录的方法【测试可用】
这篇文章主要介绍了Python实现遍历目录的方法,涉及Python针对目录与文件的遍历、判断、读取相关操作技巧,需要的朋友可以参考下
2017-03-03
Python中Numpy包的安装与使用方法简明教程
这篇文章主要介绍了Python中Numpy包的安装与使用方法,结合简单实例形式分析了Python使用pip命令在线与离线whl包安装,以及使用numpy打印随机数矩阵的操作技巧,需要的朋友可以参考下
2018-07-07
Python如何import文件夹下的文件(实现方法)
下面小编就为大家带来一篇Python如何import文件夹下的文件(实现方法)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2017-01-01
Python实现数值积分方式
今天小编就为大家分享一篇Python实现数值积分方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-11-11