Lightweight and robust ship detection method driven by self-attention mechanism
[Objective]It 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 per-formance through the design of a dedicated backbone network,improved feature extraction process,and con-strained microscopic detection heads.[Method]First,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 environ-ment,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 fur-ther 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 ro-bustness.[Results]To validate the proposed method,a dataset called 2023ships is established which con-sists 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 perform-ance compared to other algorithms,with a mAP of 92.9%,a precision rate of 92.1%,and a parameter size of 35 366 310.[Conclusion]The proposed method can greatly benefit the fields of maritime monitoring and intelligent navigation.
万方数据
维普
