Lightweight and robust ship detection method driven by self-attention mechanism

Abstract

ObjectiveIt is vital to detect and track ships during coastal monitoring and ship navigation over long distances in complex circumstances. However, due to their small size and unclear features, they can be readily confused with shorelines, noise, and rocks, making them sometimes difficult to spot immediately. To address this issue, a novel ship detection method called ShipDet is proposed which significantly improves performance through the design of a dedicated backbone network, improved feature extraction process, and constrained microscopic detection heads. MethodFirst, this method constructs a feature fusion and extraction network that is highly sensitive to small objects by integrating the Swin Transformer module (STR) with the classic CSPDarknet53 network. This enhances the correlation between small target features and the environment, establishing associations between ships and waterways, ships and other ships, and ships and coastlines, while suppressing irrelevant information. Subsequently, considering the uneven distribution and minor scale variations of ship targets in the dataset, two detection layers are retained to reduce model parameters and further enhance model performance. Moreover, the method employs the SCYLLA-IoU (SIoU) loss function to constrain the detection heads, thereby reducing regression freedom and improving detection accuracy and robustness. ResultsTo validate the proposed method, a dataset called 2023ships is established which consists of up to 9 000 samples covering various scenarios such as inland rivers, coastal areas, daytime, nighttime, and foggy weather. During testing, the proposed method demonstrates superior overall ship detection performance compared to other algorithms, with a mAP of 92.9%, a precision rate of 92.1%, and a parameter size of 35 366 310. ConclusionThe proposed method can greatly benefit the fields of maritime monitoring and intelligent navigation.

Keywords

• ship detection
• complex scenarios
• swin transformer
• siou