关键词:
深度学习
海洋生物
目标检测
Jetson TX2
摘要:
海洋生物检测与定位,对海洋资源合理开发与利用,具有非常重要的研究意义和研究价值。海洋场景复杂,传统目标检测方法无法满足检测要求。本文以深度学习相关技术为研究工具,以海洋生物(海胆、海参、海星和扇贝)为研究对象,建立并训练海洋生物目标检测模型,本文主要工作如下:
针对目标特征提取困难、海洋场景复杂和目标多尺度等问题,提出一种基于并行高分辨率网络的海洋生物目标检测算法。首先,针对目标特征难以提取的问题,采用更轻量级高分辨率网络,增强目标特征表达,有效减少图像在采样过程中损失的语义信息。然后,针对背景复杂的问题,改进注意力模块,增强在空间和通道维度的特征,捕获不规则和详细的对象布局。最后,针对目标多尺度的问题,构建感受野增强模块进一步获得足够的语义信息和丰富的细节信息,提高对多尺度目标的检测能力。实验结果显示,该方法在URPC2020、URPC2018和PASCAL VOC2007三个数据集上m AP分别达到了81.17%、77.02%和82.9%。并将模型部署到嵌入式平台Jetson TX2上。针对Jetson TX2平台,模型在URPC2020数据集上的m AP为80.64%,检测速度为1.20FPS。
针对小目标、遮挡和模糊等问题,提出一种基于引导特征学习的海洋生物目标检测算法。首先,针对语义特征丢失的问题,构建特征提取网络,以增强网络对小目标信息的提取能力。然后,针对特征分布分散的问题,设计上下文增强模块融入特征金字塔网络,以丰富上下文信息。接着,针对小目标模糊及前背景区分不明显的问题,构建引导特征学习模块,过滤冲突信息,增强目标本身的可辨性。最后,针对检测任务冲突的问题,提出一种并行交互检测头,通过引入特征交互思想与解耦检测头,来加强对目标的定位与分类能力。实验结果显示,该方法在URPC2020、URPC2018和PASCAL VOC2007三个数据集上m AP分别达到了83.46%、78.86%和83.2%,比基于并行高分辨率网络的海洋生物目标检测算法的m AP分别高了2.29%、1.84%和0.3%。并在嵌入式平台Jetson TX2上,对模型进行部署,可以实现对海洋生物图像的检测应用,模型在URPC2020数据集上的m AP为83.07%,检测速度为0.90FPS。