Affiliation:
1. College of Field Engineering, Army Engineering University of PLA, Nanjing 210007, China
Abstract
UAV remote sensing (RS) image object detection is a very valuable and challenging technology. This article discusses the importance of key features and proposes an object detection network (URSNet) based on a bidirectional multi-span feature pyramid and key feature capture mechanism. Firstly, a bidirectional multi-span feature pyramid (BMSFPN) is constructed. In the process of bidirectional sampling, bicubic interpolation and cross layer fusion are used to filter out image noise and enhance the details of object features. Secondly, the designed feature polarization module (FPM) uses the internal polarization attention mechanism to build a powerful feature representation for classification and regression tasks, making it easier for the network to capture the key object features with more semantic discrimination. In addition, the anchor rotation alignment module (ARAM) further refines the preset anchor frame based on the key regression features extracted by FPM to obtain high-quality rotation anchors with a high matching degree and rich positioning visual information. Finally, the dynamic anchor optimization module (DAOM) is used to improve the ability of feature alignment and positive and negative sample discrimination of the model so that the model can dynamically select the candidate anchor to capture the key regression features so as to further eliminate the deviation between the classification and regression. URSNet has conducted comprehensive ablation and SOTA comparative experiments on challenging RS datasets such as DOTA-V2.0, DIOR and RSOD. The optimal experimental results (87.19% mAP, 108.2 FPS) show that URSNet has efficient and reliable detection performance.
Funder
National Natural Science Foundation of China
Reference87 articles.
1. A Method for Rapidly Determining the Seismic Performance of Buildings Based on Remote-Sensing Imagery and Its Application;Yu;Adv. Civ. Eng.,2022
2. Real-Time Garbage Object Detection with Data Augmentation and Feature Fusion Using SUAV Low-Altitude Remote Sensing Images;Chen;IEEE Geosci. Remote Sens. Lett.,2022
3. UPDExplainer: An interpretable transformer-based framework for urban physical disorder detection using street view imagery;Hu;ISPRS J. Photogramm. Remote Sens.,2023
4. Uncertainty-aware convolutional neural network for explainable artificial intelligence-assisted disaster damage assessment;Cheng;Struct. Control Health Monit.,2022
5. Military Target Detection in Remote Sensing Imagery Based on YOLOv4-Faster;Wang;J. Imaging Sci. Technol.,2022