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()
相关推荐
xingshanchang33 分钟前
PyTorch 不支持旧GPU的异常状态与解决方案:CUDNN_STATUS_NOT_SUPPORTED_ARCH_MISMATCH
人工智能·pytorch·python
费弗里4 小时前
Python全栈应用开发利器Dash 3.x新版本介绍(1)
python·dash
李少兄9 天前
解决OSS存储桶未创建导致的XML错误
xml·开发语言·python
就叫飞六吧9 天前
基于keepalived、vip实现高可用nginx (centos)
python·nginx·centos
Vertira9 天前
PyTorch中的permute, transpose, view, reshape和flatten函数详解(已解决)
人工智能·pytorch·python
学Linux的语莫9 天前
python基础语法
开发语言·python
匿名的魔术师9 天前
实验问题记录:PyTorch Tensor 也会出现 a = b 赋值后,修改 a 会影响 b 的情况
人工智能·pytorch·python
Ven%9 天前
PyTorch 张量(Tensors)全面指南:从基础到实战
人工智能·pytorch·python
mahuifa9 天前
PySide环境配置及工具使用
python·qt·环境配置·开发经验·pyside
大熊猫侯佩9 天前
ruby、Python 以及 Swift 语言关于 “Finally” 实现的趣谈
python·ruby·swift