ffmpeg与opencv实现rtsp推流(基于ZLMediaKit实现)

前言

之前记录过，GStreamer与opencv实现rtsp推流实现，这次通过ffmpeg与opencv并且搭配ZLMediaKit实现rtsp推流。

代码

import cv2
import time
import subprocess as sp
import multiprocessingclass stream_pusher(object):def __init__(self, rtmp_url=None, raw_frame_q=None): # 类实例化的时候传入rtmp地址和帧传入队列self.rtmp_url = rtmp_urlself.raw_frame_q = raw_frame_qfps = 20 # 设置帧速率# 设置分辨率width = 1920 # 宽height = 1080 # 高# 设置FFmpeg命令文本self.command = [&＃39;ffmpeg&＃39;,&＃39;-y&＃39;,&＃39;-f&＃39;, &＃39;rawvideo&＃39;,&＃39;-vcodec&＃39;, &＃39;rawvideo&＃39;,&＃39;-pix_fmt&＃39;, &＃39;bgr24&＃39;,&＃39;-s&＃39;, "{}x{}".format(width, height),&＃39;-r&＃39;, str(fps),&＃39;-i&＃39;, &＃39;-&＃39;,&＃39;-c:v&＃39;, &＃39;libx264&＃39;,&＃39;-pix_fmt&＃39;, &＃39;yuv420p&＃39;,&＃39;-preset&＃39;, &＃39;ultrafast&＃39;,&＃39;-f&＃39;, &＃39;flv&＃39;,self.rtmp_url]# 向服务器推送def push_frame(self):# 配置向os传递命令的管道p = sp.Popen(self.command, stdin=sp.PIPE)while True:if not self.raw_frame_q.empty(): # 如果输入管道不为空# 把帧和相关信息从输入队列中取出frame = self.raw_frame_q.get()# 把内容放入管道，放入后有os自己去执行p.stdin.write(frame.tostring())else:time.sleep(0.01)# 启动运行def run(self):# 定义一个子进程push_frame_p = multiprocessing.Process(target=self.push_frame, args=())push_frame_p.daemon = True # 把子进程设置为daemon方式push_frame_p.start() # 运行子进程if __name__ == &＃39;__main__&＃39;:cap = cv2.VideoCapture("rtsp://admin:admin110@192.168.2.64:554/main")rtmpUrl = "rtmp://192.168.2.8/live/test666" # 用vcl等直播软件播放时，也用这个地址raw_q = multiprocessing.Queue() # 定义一个向推流对象传入帧及其他信息的队列my_pusher = stream_pusher(rtmp_url=rtmpUrl, raw_frame_q=raw_q) # 实例化一个对象my_pusher.run() # 让这个对象在后台推送视频流while True:_, raw_frame = cap.read()if not raw_q.full(): # 如果队列没满raw_q.put(raw_frame) # 送入队列cv2.waitKey(1)cap.release()print(&＃39;finish&＃39;)

效果如下：

总结

工作中总会遇到各种问题，解决一个记录一个，向着更优秀的程序员迈进！！！继续加油喽！！！
如果阅读本文对你有用，欢迎一键三连呀！！！
2021年8月31日20:38:34