python利用ffmpeg进行rtmp推流直播

电商运营 2022-12-11 21:56 11287人围观推流

思绪：

opencv读取视频 —> 将视频朋分为帧 —> 将每一帧停止需求加工后 —> 将此帧写入pipe管道 —> 操纵ffmpeg停止推流直播

pipe管道：

啥是pipe管道？大略的了解就是一个放同享文件的地方（了解不是很深入。。。）操纵这个特点, 把处置后的图片放入管道, 让ffmpeg读取处置后的图像帧并停止rtmp推流即可

代码

import subprocess as sp

rtmpUrl = ""
camera_path = ""
cap = cv.VideoCapture(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
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',
rtmpUrl]

# 管道设置
p = sp.Popen(command, stdin=sp.PIPE)

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

# process frame
# your code
# process frame

# write to pipe
p.stdin.write(frame.tostring())

代码说明:

rtmpUrl就是要接收视频的办事器了, 我做尝试时是在自己机子上设置了一个nginx办事器接收视频流(ubuntu 不要经过apt安装哦, 请从源码安装, 由于apt安装的版本没有rtmp协议, 需要下载nginx源码然后配合nginx-rtmp-module这个工具安装保举一篇nginx安装教程)
camera_path就是要停止直播的视频地址了
重点的代码实在就这几句:

import subprocess as sp

# ffmpeg command
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',
rtmpUrl]

# 管道设置
p = sp.Popen(command, stdin=sp.PIPE)

# write to pipe
p.stdin.write(frame.tostring())

我读取的视频是rtsp收集摄像头的视频流, 可是一旦运转没多久就会出现 pipe broke 的报错(1080p视频), 不晓得是什么缘由, 若有大神还请指导指导

WriteN, RTMP send error 104 (129 bytes)

更新: 我改1080p为720p 临时没出现这个题目

我在读取视频时辰还碰到这个报错

error while decoding xxx

经过一方百度, 是“FFMPEG Lib对在rtsp协议中的H264 vidos不支持”的题目, 处理方式: 法式开启两个线程, 一个线程读取摄像头的帧, 另一个线程处置这帧图片, 这里还保举一个大佬用行列处置视频的方式 ,大师可以套用一下啊飞机票 多线程处置的版本：

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]