python显示rtsp视频，并实时框人员 推送给流媒体服务器_cv2.videowriter rtmp

要在Python中显示RTSP视频流、实时框出画面中的人员，并将带有框的视频推送至流媒体服务器，可以分为以下几个步骤：

步骤一：获取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

步骤二：推送带有框的视频至流媒体服务器

这里以使用opencv-python内置的cv2.VideoWriter类将处理后的视频帧写入RTMP流为例。请确保您使用的OpenCV版本支持RTMP输出，并且流媒体服务器（如Nginx-RTMP、Wowza、OBS等）已配置好接收RTMP流的地址。

python

import cv2
 
rtsp_url = 'rtsp://your_rtsp_stream_url'
rtmp_output_url = 'rtmp://your_rtmp_server_address/stream_key'
 
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')
 
# 初始化VideoWriter，指定编码、帧率、分辨率和输出URL
fourcc = cv2.VideoWriter_fourcc(*'FLV1')  # 或根据实际情况选择其他编码器
fps = cap.get(cv2.CAP_PROP_FPS)
frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
out = cv2.VideoWriter(rtmp_output_url, fourcc, fps, (frame_width, frame_height))
 
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)
 
    # 将处理后的帧写入RTMP流
    out.write(frame)
 
    # 显示带有框的帧（可选，仅用于本地调试）
    cv2.imshow('RTSP Video Stream with Person Detection', frame)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break
 
# 释放资源
cap.release()
out.release()
cv2.destroyAllWindows()

在这个示例中：

1. 使用cv2.VideoWriter创建一个RTMP输出对象，指定编码器、帧率、分辨率和RTMP输出URL。
2. 在主循环中，读取视频帧，进行人员检测，绘制矩形框，然后将处理后的帧写入out对象（即RTMP流）。
3. 本地调试时可选择显示带有框的帧，按'q'键退出。

请注意，实际使用时需要根据您的流媒体服务器设置调整rtmp_output_url。如果您的OpenCV版本不支持RTMP输出，或者需要更灵活的流推送方式，可以考虑使用第三方库如ffmpeg-python或streamlink结合ffmpeg命令行工具进行流推送。