Statistical Adaptation Loss Improved SMALL Sample Ship Detection Method Based on an Attention Mechanism and Data Enhancement

Abstract

Synthetic aperture radar (SAR) imagery is a promising data source for ocean activity detection. Ship target detection based on SAR images is widely used in maritime trade and the military. SAR image data are rare, and the amount of public data is small. For applications of SAR image ship target detection, a model with low data dependence, fast iteration and low training cost is needed. In this paper, the balanced positive and negative data enhancement method was used. Through statistical analysis of the training dataset, similar sea areas in the training set are filled with detection targets with comfortable size features. Increasing the proportion of positive samples in the data helps to improve the model detection effect. The regional attention preadaptation mechanism based on statistical analysis was implemented to extract information, and the scale-adaptive loss was combined to improve the detection accuracy of the model. Using the same data, our model exhibited better performance. When using 30% of the data, our model was stable in terms of accuracy and average precision (AP) and maintained detection results similar to the training results achieved using 100% of the dataset.