Abstract:This paper presents an enhanced ship detection algorithm for YOLOv7‐tiny, addressing occlusion challenges in multi‐target ship detection. The modifications include integrating Convolutional Block Attention Module for position emphasis, GSConv in the Neck section, Content‐Aware ReAssembly of Features for up‐sampling, and SIOU Loss for better accuracy. The improved model achieves a precision of 91.2%, mAP@0.5 of 94.5%, and an F1‐score of 90.7%, surpassing the original model by 3.7%, 5.5%, and 4.2%, respectively, significantly boosting detection accuracy. In addressing the challenge of ships being prone to occlusion in multi‐target situations during ship target detection, leading to missed and false detections, this paper proposes an enhanced ship detection algorithm for YOLOv7‐tiny. The proposed method incorporates several key modifications. Firstly, it introduces the Convolutional Block Attention Module in the Backbone section of the original model, emphasizing position information while attending to channel features to enhance the network's ability to extract crucial information. Secondly, it replaces standard convolution with GSConv convolution in the Neck section, preserving detailed information and reducing computational load. Subsequently, the lightweight operator Content‐Aware ReAssembly of Features is employed to replace the original nearest‐neighbour interpolation, mitigating the loss of feature information during the up‐sampling process. Finally, the localization loss function, SIOU Loss, is utilized to calculate loss, expedite training convergence, and enhance detection accuracy. The research results indicate that the precision of the improved model is 91.2%, mAP@0.5 is 94.5%, and the F1‐score is 90.7%. These values are 3.7%, 5.5%, and 4.2% higher than those of the original YOLOv7‐tiny model, respectively. The improved model effectively enhances detection accuracy. Additionally, the improved model achieves an FPS of 145.4, meeting real‐time requirements.

Ship detection based on Improved YOLO Algorithm

Multi-Scale Ship Detection from SAR and Optical Imagery Via A More Accurate YOLOv3

Ship Target Detection Algorithm Based on Improved YOLOX_s

High-Efficiency and High-Precision Ship Detection Algorithm Based on Improved YOLOv8n

An Improved YOLO V3 Small-Scale Ship Target Detection Algorithm

Research on Improved Lightweight YOLOv5s for Multi-Scale Ship Target Detection

Ship Target Detection Algorithm Based on Improved YOLOv5

Marine Target Detection Algorithm Based on Deep Learning

Ship Detection Algorithm based on Improved YOLO V5

ShipYOLO: an Enhanced Model for Ship Detection

YOLOv7-ship: an Efficient Method for Marine Ship Detection Based on Wave Glider.

Research on the Automatic Detection of Ship Targets Based on an Improved YOLO v5 Algorithm and Model Optimization

Ship target detection algorithm of optical remote sensing image based on YOLOv5

Optical Remote Sensing Ship Recognition and Classification Based on Improved YOLOv5

Improved YOLOv3 Based on Attention Mechanism for Fast and Accurate Ship Detection in Optical Remote Sensing Images

Ship Object Detection and Classification Based on Improved YOLOv5

Ship Detection: An Improved YOLOv3 Method

An Efficient Enhanced-YOLOv5 Algorithm for Multi-scale Ship Detection

Improvement of ship target detection algorithm for YOLOv7‐tiny

An Improved YOLO-v3Algorithm for Ship Detection in SAR Image Based on K-means++ with Focal Loss

Enhanced YOLO v3 Tiny Network for Real-Time Ship Detection From Visual Image