Real-Time Long-Distance Ship Detection Architecture Based on YOLOv8

Abstract

Long-distance detection of maritime ships is pivotal for the development of intelligent collision avoidance systems. Despite significant advancements in target detection achieved through deep learning, the identification of long-distance ships poses a substantial challenge due to their small pixel size in images. Consequently, the recognition of long-distance ships essentially amounts to small object detection. In response to these challenges in small object detection, this paper proposes Ship-YOLOv8, a modified architecture derived from You Only Look Once version 8 (YOLOv8). First, we developed the C-Bottleneck Transformer neural network (C-BoTNet), which is integrated at the end of the backbone, to enhance the global receptive field and facilitate feature fusion. Additionally, we incorporated shallow features with deep features and introduced a dedicated detection layer for small objects into the original structure. Furthermore, we optimized the C2f in the neck using the cross stage partial network (VoVGSCSP) based on GSConv. Finally, we conducted optimization using the Wise-IoU loss function. Extensive experiments conducted on a self-created dataset of long-distance ships demonstrate the remarkable capabilities of Ship-YOLOv8. The proposed method achieves an AP0.5 of 91.8%, significantly outperforming YOLOv8’s AP0.5 of 70.6%. Moreover, our method attains a detection speed of 4.8 ms per image during inference, showcasing its efficiency in real-time applications. To validate the algorithm’s broad applicability, comparative experiments were conducted on a public maritime dataset SeaShips. Ship-YOLOv8 achieved an AP0.5 score of 99.3%, surpassing YOLOv8’s 98.6%. Code is available at https://github.com/zihaohao123/Ship-YOLOv8.

Keywords

• Long-distance ship
• small object detection
• deep learning
• YOLOv8