103 lines
4.0 KiB
Python
103 lines
4.0 KiB
Python
# 进行摄像头直播推流 / live_broadcast
|
||
|
||
import cv2 as cv
|
||
import time
|
||
import subprocess as sp
|
||
import multiprocessing
|
||
import platform
|
||
import psutil
|
||
import datetime
|
||
|
||
|
||
class stream_pusher(object):
|
||
def __init__(self, rtmp_url=None, raw_frame_q=None): # 类实例化的时候传入rtmp地址和帧传入队列
|
||
self.rtmp_url = rtmp_url
|
||
self.raw_frame_q = raw_frame_q
|
||
|
||
fps = 35 # 设置帧速率
|
||
# 设置分辨率
|
||
width = 640 # 宽
|
||
height = 480 # 高
|
||
|
||
# 设置FFmpeg命令文本
|
||
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', 'slow',
|
||
'-f', 'flv',
|
||
self.rtmp_url]
|
||
|
||
# 对获取的帧做一些画面处理的方法，返回完成处理的帧。
|
||
def __frame_handle__(self, raw_frame, text, shape1, shape2):
|
||
# 帧用cv2进行一些处理，比如写上文本，画矩形等
|
||
cv.putText(raw_frame, text.split('.')[0], (20, 40), cv.FONT_ITALIC, 0.8, (0, 0, 255), 1, cv.LINE_AA)
|
||
return (raw_frame)
|
||
|
||
# 向服务器推送
|
||
def push_frame(self):
|
||
# 指定在哪些cpu核上运行
|
||
p = psutil.Process()
|
||
p.cpu_affinity([0, 1, 2, 3])
|
||
# 配置向os传递命令的管道
|
||
p = sp.Popen(self.command, stdin=sp.PIPE)
|
||
|
||
while True:
|
||
# if not self.raw_frame_q.empty(): # 如果输入管道不为空
|
||
# 把帧和相关信息从输入队列中取出
|
||
raw_frame, text, shape1, shape2 = self.raw_frame_q.get()
|
||
# 对获取的帧进行画面处理
|
||
frame = self.__frame_handle__(raw_frame, text, shape1, shape2)
|
||
|
||
# 把内容放入管道，放入后有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__ == '__main__':
|
||
# 根据不同的操作系统，设定读取哪个摄像头
|
||
if platform.system() == 'Linux': # 如果是Linux系统
|
||
cap = cv.VideoCapture(10) # 绑定编号为10的摄像头
|
||
cap.set(cv.CAP_PROP_FOURCC, cv.VideoWriter_fourcc('M', 'J', 'P', 'G'))
|
||
cap.set(3, 640) # 设置摄像头画面的宽
|
||
cap.set(4, 480) # 设置摄像头画面的高
|
||
elif platform.system() == 'Darwin': # 如果是苹果的OS X系统
|
||
cap = cv.VideoCapture(0) # 绑定编号为0的摄像头
|
||
cap.set(cv.CAP_PROP_FOURCC, cv.VideoWriter_fourcc('M', 'J', 'P', 'G'))
|
||
cap.set(3, 640)
|
||
cap.set(4, 480)
|
||
else: # 没有windows系统，所以就不判断了
|
||
cap = cv.VideoCapture(0) # 绑定编号为0的摄像头
|
||
cap.set(cv.CAP_PROP_FOURCC, cv.VideoWriter_fourcc('M', 'J', 'P', 'G'))
|
||
# cap.set(3, 640)
|
||
# cap.set(4, 480)
|
||
|
||
rtmpUrl = "rtmp://39.97.124.237:1984/wodelive" # 用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()
|
||
info = (raw_frame, str(datetime.datetime.now()), '3', '4') # 把需要送入队列的内容进行封装
|
||
# if not raw_q.full(): # 如果队列没满
|
||
raw_q.put(info) # 送入队列
|
||
cv.waitKey(1)
|
||
cap.release()
|
||
print('finish')
|