LKPF-YOLO: A Small Target Ship Detection Method for Marine Wide-Area Remote Sensing Images

Ship detection based on wide-area remote sensing imagery has a wide range of applications in areas such as ship supervision and rescue at sea. However, wide-area remote sensing satellites sacrifice spatial resolution and spectral resolution to cover a larger sea area, which leads to smaller ship scales, fewer source pixels, and a lack of texture details in the images. In this paper, we propose a deep learning network, LKPF-YOLO, for detecting small-target ships in wide-area remote sensing images. For this purpose, we first create a South China Sea wide-area remote sensing dataset containing about 7600 ship instances. In order to extract features of small objects and low-contrast targets more efficiently, we design a re-parameterized large kernel module, C2Rep, to give the network a larger effective sensing field and richer gradient flow information. Finally, we design a loss function, Priori Focal Loss, based on unbalanced learning and prior knowledge, which guides the model to focus more on the training of small and difficult samples. The experimental results show that the model achieves accurate and stable small-target ship detection in wide-area remote sensing datasets. The mAP(50) (mean Average Precision) and mAP(50:95) of the model reached 93.6% and 50.7%, which were 5.5% and 12.9% higher than the original model, respectively. In addition, the number of parameters and computation of the model are reduced by 7% and 18.7%, respectively, providing great potential for model deployment.