ERF-RTMDet: An Improved Small Object Detection Method in Remote Sensing Images
-
Published:2023-11-30
Issue:23
Volume:15
Page:5575
-
ISSN:2072-4292
-
Container-title:Remote Sensing
-
language:en
-
Short-container-title:Remote Sensing
Author:
Liu Shuo1ORCID, Zou Huanxin1ORCID, Huang Yazhe2, Cao Xu1, He Shitian1, Li Meilin1ORCID, Zhang Yuqing1
Affiliation:
1. College of Electronic Science and Technology, National University of Defense Technology, Changsha 410073, China 2. Tianjin Advanced Technology Research Institute, Tianjin 300457, China
Abstract
A significant challenge in detecting objects in complex remote sensing (RS) datasets is from small objects. Existing detection methods achieve much lower accuracy on small objects than medium and large ones. These methods suffer from limited feature information, susceptibility to complex background interferences, and insufficient contextual information. To address these issues, a small object detection method with the enhanced receptive field, ERF-RTMDet, is proposed to achieve a more robust detection capability on small objects in RS images. Specifically, three modules are employed to enhance the receptive field of small objects’ features. First, the Dilated Spatial Pyramid Pooling Fast Module is proposed to gather more contextual information on small objects and suppress the interference of background information. Second, the Content-Aware Reassembly of Features Module is employed for more efficient feature fusion instead of the nearest-neighbor upsampling operator. Finally, the Hybrid Dilated Attention Module is proposed to expand the receptive field of object features after the feature fusion network. Extensive experiments are conducted on the MAR20 and NWPU VHR-10 datasets. The experimental results show that our ERF-RTMDet attains higher detection precision on small objects while maintaining or slightly enhancing the detection precision on mid-scale and large-scale objects.
Funder
Natural Science Foundation of China
Subject
General Earth and Planetary Sciences
Reference46 articles.
1. Lai, J., Liu, C.L., Chen, X., Zhou, J., Tan, T., Zheng, N., and Zha, H. (2018, January 9–12). Pattern Recognition and Computer Vision. Proceedings of the Lecture Notes in Computer Science, Guildford, UK. 2. Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks;Ren;IEEE Trans. Pattern Anal. Mach. Intell.,2017 3. Redmon, J., Divvala, S.K., Girshick, R.B., and Farhadi, A. (July, January 26). You Only Look Once: Unified, Real-Time Object Detection. Proceedings of the 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Las Vegas, NV, USA. 4. Redmon, J., and Farhadi, A. (2017, January 21–26). YOLO9000: Better, Faster, Stronger. Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), Honolulu, HI, USA. 5. Redmon, J., and Farhadi, A. (2018). YOLOv3: An Incremental Improvement. arXiv.
|
|