python使用onvif连接rtsp协议的摄像头

原创

mob64ca12f3f05d 2024-09-28 04:56:35 ©著作权

文章标签 python 设备信息视频流 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12f3f05d的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用 ONVIF 连接 RTSP 协议摄像头的指南

当你刚入门开发并需要连接 RTSP 协议的摄像头时，很多人会感到迷茫。本文将带你一步一步实现使用 Python 和 ONVIF 连接 RTSP 摄像头的过程。首先，我们将介绍整个过程的步骤，接着告诉你每一步需要做什么，并给出相应的代码示例。

流程步骤

以下是实现 ONVIF 连接 RTSP 摄像头的基本步骤：

步骤	描述
1. 安装相关库	安装 `zeep` 和 `onvif_zeep` 库
2. 连接摄像头	使用 ONVIF 协议连接到摄像头
3. 获取设备信息	获取摄像头的基本信息
4. 获取流 URL	获取视频流的 RTSP URL
5. 连接流	使用获取的 RTSP URL 连接到摄像头视频流

详细步骤

步骤 1：安装相关库

在开始编码之前，你需要确保已安装 onvif_zeep 库。这是一个实现 ONVIF 协议的 Python 库。你可以通过以下命令来安装：

pip install onvif_zeep

步骤 2：连接摄像头

在此步骤中，我们将使用 ONVIF 协议连接到摄像头，并验证连接是否成功。创建一个 Python 文件，并添加以下代码：

from onvif import ONVIFCamera

# 定义摄像头的 IP 地址、用户名和密码
ip = '192.168.1.100'
port = 80  # 默认 ONVIF 端口
username = 'admin'
password = 'password'

# 初始化 ONVIFCamera
camera = ONVIFCamera(ip, port, username, password)

# 测试连接（如果连接成功，应该不会报错）
try:
    print("Successfully connected to camera.")
except Exception as e:
    print(f"Failed to connect to camera: {e}")

代码解释：

ONVIFCamera类用于实例化一个摄像头对象。你需要传入摄像头的 IP 地址、端口、用户名和密码。
print 语句用于反馈连接状态。

步骤 3：获取设备信息

连接成功后，你可以通过以下代码获取摄像头的基本信息：

# 获取设备信息
device_service = camera.create_devicemgmt_service()
device_info = device_service.GetDeviceInformation()

# 打印设备信息
print(f"Model: {device_info.Model}")
print(f"Manufacturer: {device_info.Manufacturer}")

代码解释：

create_devicemgmt_service() 创建设备管理服务的实例。
GetDeviceInformation() 方法返回摄像头的基本信息。

步骤 4：获取流 URL

接下来，我们需要获取 RTSP 流的 URL。这可以通过以下代码实现：

# 获取媒体服务
media_service = camera.create_media_service()

# 获取监视配置
profiles = media_service.GetProfiles()

# 获取流的 RTSP URL
stream_uri = media_service.GetStreamUri({'StreamSetup': {'Stream': 'RTP-Unicast', 'Transport': {'Protocol': 'RTSP'}}, 'ProfileToken': profiles[0].token})

# 打印 RTSP URL
print(f"RTSP Stream URI: {stream_uri.Uri}")

代码解释：

GetProfiles() 方法获取摄像头的所有媒体配置文件。
GetStreamUri() 方法返回流的 URI，将第一项配置文件传入以获取其流地址。

步骤 5：连接流

最后一步是连接到 RTSP 流。可以使用 opencv-python 库来完成这一操作。安装库：

pip install opencv-python

然后使用以下代码连接并播放视频流：

import cv2

# 使用 OpenCV 连接 RTSP 流
cap = cv2.VideoCapture(stream_uri.Uri)

while True:
    ret, frame = cap.read()
    if not ret:
        print("Failed to grab frame.")
        break

    cv2.imshow("Camera Feed", frame)

    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()

代码解释：

cv2.VideoCapture 用于打开 RTSP 流。
while 循环读取并显示摄像流，按下 'q' 键退出视频窗口。

类图

下面是类图的 mermaid 语法表示：

classDiagram
class ONVIFCamera {
    +string ip
    +int port
    +string username
    +string password
    +create_devicemgmt_service()
    +create_media_service()
}

class DeviceService {
    +GetDeviceInformation()
}

class MediaService {
    +GetProfiles()
    +GetStreamUri(streamSetup, profileToken)
}

ONVIFCamera --> DeviceService : Creates
ONVIFCamera --> MediaService : Creates