python接入rtsp视频,并实时框人员 推送给rtsp流媒体服务器
创始人
2025-01-07 17:08:11
0

要实现Python显示RTSP视频流、实时框出画面中的人员,并将带有框的视频以RTSP格式推送给流媒体服务器,您需要使用支持RTSP推流的库或工具。由于OpenCV的cv2.VideoWriter并不直接支持RTSP推流,这里推荐使用ffmpeg作为中介,结合ffmpeg-python库来实现这一功能。以下是详细的步骤:

步骤一:获取RTSP视频流并进行人员检测

使用OpenCV读取RTSP视频流并进行人员检测(这里仍以人脸检测为例,您可以替换为其他人员检测模型)。同时,为了将检测结果叠加到原视频流上,我们需要保留原视频帧和检测到的人员框信息:

 

python

import cv2  rtsp_url = 'rtsp://your_rtsp_stream_url' cap = cv2.VideoCapture(rtsp_url)  if not cap.isOpened():     print("Could not open RTSP stream.")     exit(-1)  face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + 'haarcascade_frontalface_default.xml')  while True:     ret, frame = cap.read()      if not ret:         print("Failed to read frame.")         break      gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)     faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))      # 存储检测到的人员框信息     person_boxes = [(x, y, x+w, y+h) for (x, y, w, h) in faces]      # 在原帧上绘制矩形框     for box in person_boxes:         cv2.rectangle(frame, (box[0], box[1]), (box[2], box[3]), (0, 255, 0), 2)      # 显示带有框的帧(可选,仅用于本地调试)     cv2.imshow('RTSP Video Stream with Person Detection', frame)     if cv2.waitKey(1) & 0xFF == ord('q'):         break

步骤二:使用ffmpeg将处理后的视频帧推送到RTSP服务器

安装ffmpegffmpeg-python库:

 

bash

pip install ffmpeg-python

接下来,使用ffmpeg-python库创建一个ffmpeg进程,将处理后的视频帧通过管道(pipe)传递给ffmpeg进行编码和RTSP推流:

 

python

import subprocess as sp import numpy as np from ffmpeg import (     Input,     Output,     pipe, )  # RTSP推流地址 rtsp_output_url = 'rtsp://your_rtmp_server_address/stream_key'  # 创建一个空的Input对象作为输入源,使用管道接收数据 input_pipe = Input(pipe='pipe:', format='rawvideo', pix_fmt='bgr24', s='{}x{}'.format(frame_width, frame_height))  # 创建RTSP输出对象 output = Output(rtsp_output_url, vcodec='libx264', pix_fmt='yuv420p', r=fps)  # 启动ffmpeg进程,连接输入管道和输出RTSP地址 ffmpeg_process = (     input_pipe     .global_args('-hide_banner')     .global_args('-loglevel', 'warning')     .output(output)     .run_async(pipe_stdin=True) )  while True:     ret, frame = cap.read()      if not ret:         print("Failed to read frame.")         break      gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)     faces = face_cascade.detectMultiScale(gray, scaleFactor=1.1, minNeighbors=5, minSize=(30, 30))      person_boxes = [(x, y, x+w, y+h) for (x, y, w, h) in faces]     for box in person_boxes:         cv2.rectangle(frame, (box[0], box[1]), (box[2], box[3]), (0, 255, 0), 2)      # 将处理后的帧送入ffmpeg进程     ffmpeg_process.stdin.write(frame.tobytes())      # 显示带有框的帧(可选,仅用于本地调试)     cv2.imshow('RTSP Video Stream with Person Detection', frame)     if cv2.waitKey(1) & 0xFF == ord('q'):         break  # 关闭输入管道,结束ffmpeg进程 ffmpeg_process.stdin.close() ffmpeg_process.wait()

在这个示例中:

  1. 使用ffmpeg-python创建一个空的Input对象,指定格式为rawvideo,像素格式为bgr24,分辨率与原视频一致。
  2. 创建一个Output对象,指定RTSP推流地址、编码器为libx264、像素格式为yuv420p,帧率为原视频帧率。
  3. 启动ffmpeg进程,连接输入管道和输出RTSP地址。
  4. 在主循环中,读取视频帧,进行人员检测,绘制矩形框,然后将处理后的帧以字节形式写入ffmpeg进程的stdin。
  5. 本地调试时可选择显示带有框的帧,按'q'键退出。

请注意,实际使用时需要根据您的流媒体服务器设置调整`rtsp_output ...问答超时,请稍后再试

相关内容

热门资讯

九分钟推荐!九游辅助辅助在哪里... 九分钟推荐!九游辅助辅助在哪里(辅助挂)果然是真的挂,扑克教程(新版有挂)-哔哩哔哩1、金币登录送、...
第2分钟神器!福州十八扑外卦,... 第2分钟神器!福州十八扑外卦,约战沙城攻略,本来真的有挂(有挂细节)-哔哩哔哩1、福州十八扑外卦透视...
3分钟分享!白金岛十胡卡辅助(... 3分钟分享!白金岛十胡卡辅助(辅助挂)都是真的有挂,新版2025教程(有挂助手)-哔哩哔哩1、完成白...
第三分钟了解!四川家园游戏辅助... 第三分钟了解!四川家园游戏辅助软件(辅助挂)一直有挂,新2025教程(有挂讲解)-哔哩哔哩四川家园游...
2分钟攻略!方片透视辅助,约局... 2分钟攻略!方片透视辅助,约局吧辅助器下载,果然有挂(有人有挂)-哔哩哔哩1、操作简单,无需注册,只...
2分钟曝光!0759湛江吴川三... 2分钟曝光!0759湛江吴川三脚鸡辅助(辅助挂)总是真的有挂,2025新版(有挂方式)-哔哩哔哩1、...
第7分钟软件!中至上饶打炸辅助... 第7分钟软件!中至上饶打炸辅助器开挂,哈灵永久辅助,真是有挂(详细教程)-哔哩哔哩1、超多福利:超高...
第2分钟解谜!河洛杠次高科技(... 第2分钟解谜!河洛杠次高科技(辅助挂)竟然真的是有挂,科技教程(果真有挂)-哔哩哔哩1、玩家可以在河...
4分钟曝光!斗城麻将微信有没有... 4分钟曝光!斗城麻将微信有没有挂(辅助挂)好像有挂,高科技教程(有挂细节)-哔哩哔哩;1、进入游戏-...
9分钟技巧!多乐找刺激窍门,顺... 9分钟技巧!多乐找刺激窍门,顺欣茶楼软件,竟然有挂(果真有挂)-哔哩哔哩1、完成多乐找刺激窍门的残局...