HFPNet: Super Feature Aggregation Pyramid Network for Maritime Remote Sensing Small-Object Detection
Chengcheng Chen, Weiming Zeng, Xiliang Zhang
Abstract
In recent years, with the development of the maritime industry, computer vision tasks based on optical remote sensing images have gained increasing attention in the field of marine remote sensing. However, in complex marine meteorological environments, traditional detection methods often suffer from high rates of missed detections and false alarms for small targets. To address this issue, this paper proposes a Super Feature Pyramid Network (HFPNet) for detecting marine remote sensing objects, which includes a Feature Enhancement Module (FEM) and a Multi-Scale Feature Aggregation Module (MFM). The FEM can highlight features of small targets and weaken complex background features in detection; the MFM can share high-level and low-level features to better fuse multi-scale feature information. Additionally, due to the lack of marine remote sensing object datasets, this paper constructs a Marine Target Detection Dataset (MTDS) containing six types of marine objects. To address the issue of imbalanced positive and negative samples in the dataset, this paper designs a network loss function to accelerate network convergence and improve the accuracy of small target detection. Compared with other models, HFPNet achieved the highest <inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"><tex-math notation="LaTeX">$mAP_{0.5}$</tex-math></inline-formula> of 97.48% and 95.46% on the self-built dataset MTDS and the public dataset NWPU-VHR-10, respectively, after environmental enhancement. At the same time, it also achieved the fastest FPS on the test set. Finally, this paper discusses the influence of attention mechanisms and post-processing methods on HFPNet and obtains the best small target detection model.