python+openCV+ffmpeg进行物联网设备拉流rtmp

前言:

通过物联网网关设备,推流的rtmp视频流,在此基础上完成对视频使用分类集的视频流抽帧(及对帧数计数,达到目标帧时才进行识别降低延迟率)检测,并重新推流为rtsp(和对rtmp处理类似,设置一个合理的ffmpeg参数)

python 复制代码
import cv2
import subprocess
import numpy as np

# RTMP流地址
rtmp_url = "rtmp:/:1935/live/"

frame_width = 720
frame_height = 480
fps = 25
# FFmpeg命令,用于拉取RTMP流并输出到管道
ffmpeg_command = [
    "ffmpeg",
    "-i", rtmp_url,
    "-g", "1",
    "-fflags", "nobuffer",  # 减少输入流的缓冲
    "-b:v", "5k",
    "-preset", "ultrafast",  # 使用快速编码预设
    "-tune", "zerolatency",  # 启用低延迟模式
    "-f", "rawvideo",
    "-pix_fmt", "bgr24",
    '-s',"{}x{}".format(frame_width, frame_height),
    "-r", str(fps),
    "-vcodec", "rawvideo",
    "-an",  # 不包含音频
    "-"
]

# 启动FFmpeg进程
ffmpeg_process = subprocess.Popen(ffmpeg_command, stdout=subprocess.PIPE, bufsize=10**3)

# 加载人脸检测器
face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')

# 视频帧的分辨率
frame_width = 720
frame_height = 480
frame_size = frame_width * frame_height * 3  # BGR格式,每个像素3个字节

while True:
    # 从FFmpeg进程读取一帧
    raw_frame = ffmpeg_process.stdout.read(frame_size)
    if not raw_frame:
        break

    # 将帧数据转换为NumPy数组
    frame = np.frombuffer(raw_frame, np.uint8).reshape((frame_height, frame_width, 3))

    # 确保帧数组可写
    frame = frame.copy()

    # 处理帧(例如人脸检测)
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    # minNeighbors表示每个候选矩形保留的邻居个数,该值越大,误检率越低,但漏检率可能会增加。
    faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=3, minSize=(10, 10),maxSize=(50, 50))
    for (x, y, w, h) in faces:
        cv2.rectangle(frame, (x, y), (x + w, y + h), (0, 255, 0), 2)

    # 显示处理后的帧
    cv2.imshow('Video Frame', frame)

    # 按下 'q' 键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 关闭进程和窗口
ffmpeg_process.stdout.close()
ffmpeg_process.terminate()
ffmpeg_process.wait()
cv2.destroyAllWindows()
相关推荐
未来之窗软件服务1 小时前
智能 IOT 设备管理系统的设计与实现 —— 从管理及售后维护视角——毕业论文——东方仙盟
物联网·仙盟创梦ide·东方仙盟
im_AMBER3 小时前
学习日志19 python
python·学习
mortimer6 小时前
安装NVIDIA Parakeet时,我遇到的两个Pip“小插曲”
python·github
@昵称不存在6 小时前
Flask input 和datalist结合
后端·python·flask
赵英英俊7 小时前
Python day25
python
东林牧之7 小时前
Django+celery异步:拿来即用,可移植性高
后端·python·django
何双新7 小时前
基于Tornado的WebSocket实时聊天系统:从零到一构建与解析
python·websocket·tornado
AntBlack8 小时前
从小不学好 ,影刀 + ddddocr 实现图片验证码认证自动化
后端·python·计算机视觉
凪卄12138 小时前
图像预处理 二
人工智能·python·深度学习·计算机视觉·pycharm
巫婆理发2228 小时前
强化学习(第三课第三周)
python·机器学习·深度神经网络