项目简介
本项目是基于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】