FGA-YOLO: A One-Stage and High-Precision Detector Designed for Fine-Grained Aircraft Recognition

Jian Wu,Fanyu Zhao,Guitao Yao,Zhonghe Jin
DOI: https://doi.org/10.1016/j.neucom.2024.129067
IF: 6
2024-01-01
Neurocomputing
Abstract:Aircraft recognition in remote sensing images (RSIs) plays a pivotal role in various applications, such as disaster relief, airport monitoring. However, various aircraft types exhibit subtle inter-class variations, and the same model can display diverse visual characteristics under varying imaging conditions, presenting a significant challenge for the fine-grained aircraft recognition. This paper proposes an efficient detection model called FGA-YOLO. Firstly, we introduce a triple feature fusion module based on feature selection (STFF) to aggregate feature maps from three different levels, thereby constructing an efficient feature pyramid network that enables the network to capture salient information hidden across different levels of feature maps. Furthermore, we design a global multi-scale gather-and-distribute module (GMGD) to inject global information into the network, enhancing the extraction of the objects’ key discriminative features. Finally, we develop EMA Slide Loss by employing exponential moving average (EMA) method to mitigate the imbalance between easy and hard samples in aircraft recognition. Extensive experiments on two aircraft recognition datasets (i.e., MAR20 and FAIR1M) demonstrate that our proposed method can significantly improve the accuracy of fine-grained aircraft recognition. Moreover, experiments on embedded devices demonstrate that the proposed FGA-YOLO is also friendly for lightweight deployment.
What problem does this paper attempt to address?