FFCA-YOLO for Small Object Detection in Remote Sensing Images

作者：Yin Zhang, Mu Ye, Gui-Yi Zhu, Yong Liu, Pengyu Guo, Junhua Yan · 发表于：IEEE Transactions on Geoscience and Remote Sensing · 年份：2024 · DOI：10.1109/tgrs.2024.3363057 · 被引用次数：533 · 研究领域：Computer Science

Issues, such as insufficient feature representation and background confusion, make detection tasks for small object in remote sensing arduous. Particularly, when the algorithm will be deployed on board for real-time processing, which requires extensive optimization of accuracy and speed under limited computing resources. To tackle these problems, an efficient detector called feature enhancement, fusion and context aware YOLO (FFCA-YOLO) is proposed in this article. FFCA-YOLO includes three innovative lightweight and plug-and-play modules: feature enhancement module (FEM), feature fusion module (FFM), and spatial context aware module (SCAM). These three modules improve the network capabilities of local area awareness, multiscale feature fusion, and global association cross channels and space, respectively, while trying to avoid increasing complexity as possible. Thus, the weak feature representations of small objects are enhanced and the confusable backgrounds are suppressed. Two public remote sensing datasets (VEDAI and AI-TOD) for small object detection and one self-built dataset (USOD) are used to validate the effectiveness of FFCA-YOLO. The accuracy of FFCA-YOLO reaches 0.748, 0.617, and 0.909 (in terms of mAP50) that exceeds several benchmark models and the state-of-the-art methods. Meanwhile, the robustness of FFCA-YOLO is also validated under different simulated degradation conditions. Moreover, to further reduce computational resource consumption while ensuring efficie...