add live function by rtmp

2020-06-29 16:45:55 +08:00
commit 0b0a0565c0
2 changed files with 110 additions and 0 deletions
--- a/Class/live/Live.html
+++ b/Class/live/Live.html
@@ -0,0 +1,18 @@
+<!DOCTYPE html>
+<html lang="en">
+
+<head>
+    <meta charset="UTF-8">
+    <title>cv</title>
+    <link href="https://vjs.zencdn.net/7.3.0/video-js.min.css" rel="stylesheet">
+    <script src="https://vjs.zencdn.net/7.3.0/video.min.js"></script>
+    <script src="https://cdn.jsdelivr.net/npm/videojs-flash@2/dist/videojs-flash.min.js"></script>
+</head>
+
+<body>
+    <video id="liveVideo" class="video-js" controls autoplay preload="auto" width="1280"
+        height="720" data-setup="{}">
+        <source src="rtmp://39.97.124.237:1984/wodelive/" type="rtmp/flv">
+    </video>
+</body>
+</html>
--- a/Class/live/RTMP_TRY.py
+++ b/Class/live/RTMP_TRY.py
@@ -0,0 +1,92 @@
+import cv2 as cv
+import time
+import subprocess as sp
+import multiprocessing
+import platform
+import psutil
+
+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 = 20  # 设置帧速率
+        # 设置分辨率
+        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',
+                        '-f', 'flv',
+                        self.rtmp_url]
+
+    # 对获取的帧做一些画面处理的方法，返回完成处理的帧。
+    def __frame_handle__(self, raw_frame, text, shape1, shape2):
+        #帧用cv2进行一些处理，比如写上文本，画矩形等
+        return(raw_frame)
+
+    # 向服务器推送
+    def push_frame(self):
+        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(3, 640) # 设置摄像头画面的宽
+        cap.set(4, 480) # 设置摄像头画面的高
+    elif platform.system() == 'Darwin': # 如果是苹果的OS X系统
+        cap = cv.VideoCapture(0) # 绑定编号为0的摄像头
+        cap.set(3, 640)
+        cap.set(4, 480)
+    else: # 没有windows系统，所以就不判断了
+        cap = cv.VideoCapture(0)  # 绑定编号为0的摄像头
+        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,'2','3','4') # 把需要送入队列的内容进行封装
+        if not raw_q.full(): # 如果队列没满
+            raw_q.put(info) # 送入队列
+        cv.waitKey(1)
+    cap.release()
+    print('finish')