基于改进YOLOv3的复杂海空场景舰船目标智能识别算法

Intelligent Recognition of Ship Targets in Complex Sea and Air Scenes Based on Improved YOLOv3 Algorithm

  • 摘要: 【目的】研究海上舰船目标检测与识别技术,对维护我国海洋权益,保护我国海洋领域安全具有重要的经济意义和战略意义。【方法】针对舰船目标较小,分布密集且存在大量的遮挡,导致舰船目标检测和识别精度低、漏检率高的问题,提出了一种基于YOLOv3的海上舰船目标检测和识别改进算法YOLOv3-ship,该算法通过优化YOLOv3的特征提取网络,增强了特征提取能力;采用Res2Net和SE-Net融合的方法,增强了算法模型的鲁棒性和泛化能力;扩展模型的输出预测尺度,有效提高了对于舰船小目标的检测和识别精度;提高舰船小目标在损失函数中的权重,进一步降低了舰船小目标的漏检率。【结果】实验结果表明YOLOv3-ship较算法的平均检测识别精度(mAP)提高9.0%。【结论】提出的YOLOv3-ship可较好解决大尺度的海空场景下,舰船目标小、分布密集、存在遮挡等情况导致的漏检率高的问题。

     

    Abstract: Objectives The research on the detection and recognition technology of marine ships has important economic and strategic significance for conservation of maritime rights and marine safety. Methods To address the problem of low accuracy and high misdetection rates caused by small targets, densely distributed targets and other occlusions; this research work proposes a novel ship target detection and recognition method named YOLOv3-ship algorithm based on YOLOv3 algorithm. The feature extraction network of the YOLOv3 algorithm has been improved to enhance the feature extraction ability. Subsequently, the fusion method of Res2Net and SE-Net is adopted to enhance the robustness and generalization ability of the algorithm model and the output prediction scale of the model is expanded to effectively improve the accuracy of detection and recognition of small ship targets. By increasing the weight for small ship targets in the loss function, the misdetection rate of small ship targets is further reduced. Results Experimental results show that the mAP based on YOLOv3-ship algorithm improved 9.0% than YOLOv3 algorithm. Conclusions The proposed YOLOv3-ship algorithm can relatively well solve the problems of high detection omission rate caused by small ship targets, dense distribution, and occlusion in large–scale maritime scenarios.

     

/

返回文章
返回