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