针对漂浮物检测任务,检测算法的输入数据可以是图像或者视频,而在水域监控中使用摄像头采集漂浮物目标的视频数据更接近实用性与实时性的检测要求,本研究所使用的DeepStream就可以较好地实时处理视频文件。NVIDIA DeepStream SDK简称DeepStream,是基于GStreamer开发的流媒体分析软件包。其形式开源并且具有结构模块化、软硬件支持全面以及流文件处理迅速的特点。 首先在计算机平台中将训练得到的.pt格式文件转化成.tws的中间格式文件。然后环境转移到Jetson Nano平台,待TensorRT相关参数修改并编译完成后,执行命令将.wts格式文件转换为.engine引擎文件,并测试是否转换成功。最后修改DeepStream相关参数文件并编译,生成可执行文件后即可运行DeepStream进行测试。网络模型在DeepStream上部署的流程如图所示。
时间: 2023-03-04 17:38:43 浏览: 175
河道漂浮物检测数据集.zip
这段文字介绍了针对漂浮物检测任务,使用摄像头采集漂浮物目标的视频数据更接近实用性与实时性的检测要求。作者使用了NVIDIA DeepStream SDK来实现实时处理视频文件,它是一个基于GStreamer开发的流媒体分析软件包,开源并且具有结构模块化、软硬件支持全面以及流文件处理迅速的特点。
作者的部署流程包括以下步骤:首先将训练得到的.pt格式文件转化成.tws的中间格式文件;然后将环境转移到Jetson Nano平台,修改TensorRT相关参数并编译完成后,将.wts格式文件转换为.engine引擎文件,并测试是否转换成功;最后修改DeepStream相关参数文件并编译,生成可执行文件后即可运行DeepStream进行测试。整个部署流程如图所示,包括网络模型在DeepStream上的部署过程。
阅读全文