Addressing issues such as small-scale targets and complex background interferences that affect the accuracy of ship detection in complex maritime scenarios, this paper proposes RSD-YOLO, an improved lightweight maritime ship detection algorithm based on YOLOv7-tiny. The method begins with the introduction of a lightweight reparameterization module called RepNCSPELAN4, and combines it with a coordinate attention mechanism to improve and obtain the RepELAN_CA structure. Using RepELAN_CA to reconstruct the backbone network enriches feature extraction and enhances target localization accuracy. By decoupling the training and inference stages, the model’s computational load and the number of parameters are significantly reduced. Subsequently, the SPD subsampling strategy, which replaces the traditional method combining striding convolution and pooling. SPD introduced to mitigate the loss of fine-grained features and enhance the detection precision of small-scale targets. Finally, DyHead is introduced to refine the original detection head, improving the model’s capability to handle complex background noise by overlaying two iterations of the DyHead attention blocks. Experimental results on the SeaShips dataset demonstrate that RSD-YOLO increases the mAP by 1.6% to 91.09%, compared to YOLOv7-tiny, and reduces the parameters and computational load by 22.8% and 32.1%, respectively. By achieving high-accuracy ship detection while reducing resource consumption in computation and memory, RSD-YOLO holds significant practical engineering value.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Lightweight Maritime Ship Object Detection Based on YOLOv7-Tiny

  • Weixiang Feng,
  • Wenbo Zhang,
  • Dongsheng Guo,
  • Zehua Jia,
  • Shan Xue

摘要

Addressing issues such as small-scale targets and complex background interferences that affect the accuracy of ship detection in complex maritime scenarios, this paper proposes RSD-YOLO, an improved lightweight maritime ship detection algorithm based on YOLOv7-tiny. The method begins with the introduction of a lightweight reparameterization module called RepNCSPELAN4, and combines it with a coordinate attention mechanism to improve and obtain the RepELAN_CA structure. Using RepELAN_CA to reconstruct the backbone network enriches feature extraction and enhances target localization accuracy. By decoupling the training and inference stages, the model’s computational load and the number of parameters are significantly reduced. Subsequently, the SPD subsampling strategy, which replaces the traditional method combining striding convolution and pooling. SPD introduced to mitigate the loss of fine-grained features and enhance the detection precision of small-scale targets. Finally, DyHead is introduced to refine the original detection head, improving the model’s capability to handle complex background noise by overlaying two iterations of the DyHead attention blocks. Experimental results on the SeaShips dataset demonstrate that RSD-YOLO increases the mAP by 1.6% to 91.09%, compared to YOLOv7-tiny, and reduces the parameters and computational load by 22.8% and 32.1%, respectively. By achieving high-accuracy ship detection while reducing resource consumption in computation and memory, RSD-YOLO holds significant practical engineering value.