python + opencv + ffmpeg 实现rtmp推流

参考

• https://zhuanlan.zhihu.com/p/103164153

步骤

需要事先安装nginx并开启rtmp服务，然后

1. read video from video use opencv
2. write video to ffmpeg with pipe

"""
python ffmpeg_rtmp.py
"""
from os import write
import subprocess as sp
import cv2 

class ffmpeg_writer:
    def __init__(self,rtmpUrl,fps,width,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', 
                '-flvflags', 'no_duration_filesize',
                rtmpUrl]
        self.pipe=sp.Popen(command, stdin=sp.PIPE)

    def write(self,frame):
        self.pipe.stdin.write(frame.tostring())

    def release(self):
        self.pipe.terminate()

if __name__ == '__main__':
    rtmpUrl = "rtmp://192.168.1.151:1935/live/av_stream"
    camera_path = "output.mp4"
    cap = cv2.VideoCapture(camera_path)

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

    # 管道配置
    writer=ffmpeg_writer(rtmpUrl,fps,width,height)
            
    # 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
        writer.write(frame)
    
    writer.release()