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

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

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

ship detectionobject detectionYOLOv7attention mechanism

李毓滦、胡秀波、李鑫军、曹睿、万占鸿、韩冰

展开 >

辽宁科技大学 机械工程与自动化学院,辽宁 鞍山 114051

浙江大学 海洋学院,浙江 舟山 316004

船舶检测 目标检测 YOLOv7 注意力机制

2024

辽宁科技大学学报
辽宁科技大学

辽宁科技大学学报

影响因子:0.349
ISSN:1674-1048
年,卷(期):2024.47(3)