基于改进YOLOv7的近岸目标船舶检测算法
A ship detection algorithm for nearshore targets based on improved YOLOv7
李毓滦 1胡秀波 1李鑫军 1曹睿 1万占鸿 2韩冰1
作者信息
- 1. 辽宁科技大学 机械工程与自动化学院,辽宁 鞍山 114051
- 2. 浙江大学 海洋学院,浙江 舟山 316004
- 折叠
摘要
为了提高近岸小目标船舶检测精度,本文提出一种基于YOLOv7网络模型的YOLO-ConSwin船舶目标检测算法,在主干网络中融合ConvNext与Swin-Transformer模块,增强模型在多尺度上捕捉特征的能力.在特征金字塔网络结构中引入SimAM无参数注意力机制,强化对重要通道特征的敏感性,增强船舶目标的权重,抑制背景噪声.实验结果表明,与YOLOv7s相比,船舶识别精确率提升11个百分点,证明YOLO-ConSwin算法满足小目标船舶检测要求.
Abstract
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.
关键词
船舶检测/目标检测/YOLOv7/注意力机制Key words
ship detection/object detection/YOLOv7/attention mechanism引用本文复制引用
出版年
2024
NETL
NSTL
万方数据