littlebot
Published on 2025-04-14 / 1 Visits
0

【源码】基于NVIDIA DeepStream SDK的物体检测模糊处理系统

项目简介

本项目是基于NVIDIA DeepStream SDK的GStreamer插件,可在视频流里检测并模糊特定对象。此插件借助CUDA加速,能快速处理NVMM内存的输入视频流,是seieric/gst-dsobjectsmosaic的改进版,具备更优化的物体掩膜功能。

项目的主要特性和功能

特性

  • CUDA加速的对象掩膜处理,所有掩膜处理在GPU上完成,保障处理快速流畅。
  • 可指定需应用模糊处理的类ID,用户能自行选择要模糊处理的对象类别。
  • 处理过程高效、流畅,得益于CUDA加速。

功能

  • 检测视频流中的特定对象。
  • 对检测到的对象应用模糊处理,保护隐私。
  • 提供可调节的掩膜参数,用户可调整最小置信度等参数以满足不同需求。

安装使用步骤

假设用户已下载本项目的源码文件: 1. 环境准备:确保满足DeepStream 6.1和OpenCV4(支持CUDA)的要求。 2. 复制项目:使用git命令复制项目到本地。 bash cd gst-dsobjectsmask 3. 安装:进入项目目录,运行make命令进行编译和安装。 bash sudo make -j$(nproc) install 4. 示例使用:下载并配置deepstream_tao_apps应用,下载所需模型,然后运行示例脚本进行测试。 - 构建自定义解析器。 - 运行示例脚本samples/peoplesegnet_test.sh。 - 生成的视频文件peoplesegnet_test.mp4将保存在当前目录。

依赖项

  • DeepStream 6.1
  • OpenCV4(支持CUDA)

注意事项

  • 该插件主要用于处理NVIDIA模型(如PeopleSegNet)的检测结果,对于其他模型可能无法正常工作。
  • 该插件仅在支持CUDA的Jetson设备上运行,不支持带有dGPU的设备。
  • 在配置和使用过程中,需要注意正确设置GPU设备ID和类ID等参数,以及确保在支持CUDA的GPU上运行。
  • 代码中包含了一些调试打印语句,生产环境中可能需要移除。

下载地址

点击下载 【提取码: 4003】【解压密码: www.makuang.net】