DeepStream 服务库 (DSL) 使用教程

2024-09-16 23:05:21作者：蔡丛锟

1. 项目介绍

DeepStream 服务库 (DSL) 是一个基于 NVIDIA® DeepStream SDK 的开源项目，旨在简化复杂的流处理管道的开发。DSL 将 NVIDIA® DeepStream 参考应用程序重新构想为一个共享库，提供了一系列的 DeepStream 管道服务。通过 DSL，开发者可以轻松构建、播放和动态修改 NVIDIA® DeepStream 管道，从而实现实时视频分析和多传感器处理。

DSL 的核心功能包括：

提供简单直观的 API 用于构建、播放和动态修改 DeepStream 管道。
支持多种源组件（如 CSI 源、V4L2 源、URI 源等）和多种输出组件（如窗口渲染、文件录制、RTSP 流等）。
支持动态添加和移除源、分支和输出组件，实现动态管道管理。
提供丰富的对象检测事件 (ODE) 服务，支持基于检测事件的自动化操作。

2. 项目快速启动

2.1 环境准备

在开始之前，请确保您的系统已经安装了以下依赖：

NVIDIA® DeepStream SDK
GStreamer
Python 3.x

2.2 安装 DSL

首先，克隆 DSL 项目到本地：

git clone https://github.com/prominenceai/deepstream-services-library.git
cd deepstream-services-library

然后，根据项目文档中的说明进行编译和安装：

make
sudo make install

2.3 快速启动示例

以下是一个简单的 Python 示例，展示如何使用 DSL 创建一个基本的 DeepStream 管道：

from dsl import *

# 创建一个 CSI 源
retval = dsl_source_csi_new('my-source', width=1280, height=720, fps_n=30, fps_d=1)

# 创建一个主推理引擎
retval += dsl_infer_gie_primary_new('my-pgie', 'path_to_config_file', 'path_to_model_engine', interval=0)

# 创建一个多源分屏器
retval += dsl_tiler_new('my-tiler', width=1280, height=720)

# 创建一个屏幕显示组件
retval += dsl_osd_new('my-osd', text_enabled=True, clock_enabled=True, bbox_enabled=True, mask_enabled=False)

# 创建一个窗口渲染组件
retval += dsl_sink_window_egl_new('my-window-sink', width=1280, height=720)

# 检查组件创建是否成功
if retval != DSL_RESULT_SUCCESS:
    print("组件创建失败")
    exit(1)

# 将组件添加到新管道
retval = dsl_pipeline_new_component_add_many('my-pipeline', ['my-source', 'my-pgie', 'my-tiler', 'my-osd', 'my-sink', None])

# 播放管道
retval = dsl_pipeline_play('my-pipeline')
if retval != DSL_RESULT_SUCCESS:
    print("管道播放失败")
    exit(1)

# 运行主循环
dsl_main_loop_run()

# 清理资源
dsl_delete_all()