Remote sensing target detection algorithm based on attention and feature alignment
To address the challenges of remote sensing images, including diverse target angles, intricate backgrounds, and overall clutter, we propose a remote sensing target detection method that utilizes attention mechanisms and feature alignment. First, a convolutional triplet attention mechanism is incorporated into the backbone network to improve the capture of contextual dependencies between adjacent regions in the feature map, thereby enhancing multi-scale feature learning from remote sensing images. Next, the detection head employs a rotational feature alignment network to create high-quality rotational anchors and aligned features for classification and coordinate regression. Finally, the Kernelized Focal IoU is used as the regression loss function for rotated bounding boxes, increasing the model's sensitivity to positioning accuracy and minimizing errors. The algorithm presented in this paper was evaluated on the large-scale remote sensing dataset DOTAv1.0, achieving an mAP of 0.786, and the experiments have demonstrated the effectiveness of the algorithm.
Authors
- Hualan Yang (ORCID: https://orcid.org/0000-0002-5990-4907)
- Mei Zhang (ORCID: https://orcid.org/0000-0001-9983-7705)
Institutions
- Guizhou University of Finance and Economics (CN)
Publication Details
- Published
- 2024-09-27
- DOI
- https://doi.org/10.1117/12.3051190
- Primary Topic
- Advanced Measurement and Detection Methods
- Type
- article
- Field-Weighted Citation Impact
- 0.00