Python —— 推流rtmp

前言

Python推流本质是调用FFmpeg的推流进程，所以" pipe.stdin.write(img.tobytes()) "这句话报错时，可以考虑是FFmpeg没有配置的原因。

 
     推流术语…省略。

 

代码

import cv2

# subprocess 模块允许我们启动一个新进程，并连接到它们的输入/输出/错误管道，从而获取返回值。
import subprocess


# 视频读取对象
cap = cv2.VideoCapture(".../xx.mp4") 

# 读取一帧
ret, frame = cap.read()

# 推流地址
rtmp = "rtmp:..."

# 推流参数
command = ['ffmpeg',
    '-y',
    '-f', 'rawvideo',
    '-vcodec','rawvideo',
    '-pix_fmt', 'bgr24',
    '-s', '720*560',			# 根据输入视频尺寸填写
    '-r', '25',
    '-i', '-',
    '-c:v', 'h264',
    '-pix_fmt', 'yuv420p',
    '-preset', 'ultrafast',
    '-f', 'flv',
    rtmp]

# 创建、管理子进程
pipe = subprocess.Popen(command, stdin=subprocess.PIPE)
size = (int(cap.get(cv2.CAP_PROP_FRAME_WIDTH)), int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT)))

# 循环读取
while cap.isOpened():
	# 读取一帧
    ret, frame = cap.read()
    if frame is None:
        print('read frame err!')
        continue
        
	# 显示一帧
    cv2.imshow("frame", frame)
	
	# 按键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

	# 读取尺寸、推流
    img=cv2.resize(frame,size)
    pipe.stdin.write(img.tobytes()) 

# 关闭窗口
cv2.destroyAllWindows()

# 停止读取
cap.release()

 
 

关注

笔者 - jxd

微信公众号搜索 “码农总动员” 或 微信扫描下方二维码，了解更多你不知道的XX，O(∩_∩)O