基于改进YOLOv7的近岸目标船舶检测算法

A ship detection algorithm for nearshore targets based on improved YOLOv7

为了提高近岸小目标船舶检测精度,本文提出一种基于YOLOv7网络模型的YOLO-ConSwin船舶目标检测算法,在主干网络中融合ConvNext与Swin-Transformer模块,增强模型在多尺度上捕捉特征的能力。在特征金字塔网络结构中引入SimAM无参数注意力机制,强化对重要通道特征的敏感性,增强船舶目标的权重,抑制背景噪声。实验结果表明,与YOLOv7s相比,船舶识别精确率提升11个百分点,证明YOLO-ConSwin算法满足小目标船舶检测要求。

In order to improve the detection accuracy of near-shore small target ships,a YOLO-ConSwin ship target detection algorithm based on YOLOv7 network model is proposed in this paper.ConvNext and Swin-Transformer modules are integrated in the trunk network to enhance the model's ability to capture features on a multi-scale.Introducing SimAM,a parameter-free attention mechanism,into the feature pyramid network structure enhances the sensitivity to important channel features,strengthens the weight of ship targets,and suppresses background noise.Experimental results show that the ship identification accuracy is improved by 11%compared to YOLOv7s,proving that the YOLO-ConSwin algorithm meets the requirements for detecting small ship targets.