RS-DETR: An Improved Remote Sensing Object Detection Model Based on RT-DETR

被引:1
|
作者
Zhang, Hao [1 ]
Ma, Zheng [2 ]
Li, Xiang [1 ]
机构
[1] China Univ Geosci, Sch Comp Sci, Wuhan 430074, Peoples R China
[2] China Univ Geosci, Informatizat Off, Wuhan 430074, Peoples R China
来源
APPLIED SCIENCES-BASEL | 2024年 / 14卷 / 22期
关键词
cascaded group attention; Focaler-GIoU; EBiFPN; object detection;
D O I
10.3390/app142210331
中图分类号
O6 [化学];
学科分类号
0703 ;
摘要
Object detection is a fundamental task in computer vision. Recently, deep-learning-based object detection has made significant progress. However, due to large variations in target scale, the predominance of small targets, and complex backgrounds in remote sensing imagery, remote sensing object detection still faces challenges, including low detection accuracy, poor real-time performance, high missed detection rates, and high false detection rates in practical applications. To enhance remote sensing target detection performance, this study proposes a new model, the remote sensing detection transformer (RS-DETR). First, we incorporate cascaded group attention (CGA) into the attention-driven feature interaction module. By capturing features at different levels, it enhances the interaction between features through cascading and improves computational efficiency. Additionally, we propose an enhanced bidirectional feature pyramid network (EBiFPN) to facilitate multi-scale feature fusion. By integrating features across multiple scales, it improves object detection accuracy and robustness. Finally, we propose a novel bounding box regression loss function, Focaler-GIoU, which makes the model focus more on difficult samples, improving detection performance for small and overlapping targets. Experimental results on the satellite imagery multi-vehicles dataset (SIMD) and the high-resolution remote sensing object detection (TGRS-HRRSD) dataset show that the improved algorithm achieved mean average precision (mAP) of 78.2% and 91.6% at an intersection over union threshold of 0.5, respectively, which is an improvement of 2.0% and 1.5% over the baseline model. This result demonstrates the effectiveness and robustness of our proposed method for remote sensing image object detection.
引用
收藏
页数:18
相关论文
共 50 条
  • [31] Optimized RT-DETR for accurate and efficient video object detection via decoupled feature aggregation
    Chen, Hao
    Huang, Wu
    Zhang, Tao
    INTERNATIONAL JOURNAL OF MULTIMEDIA INFORMATION RETRIEVAL, 2025, 14 (01)
  • [32] LSH-DETR: object detection algorithm for marine benthic organisms based on improved DETR
    Wu, Shuang
    Liu, Yong
    Li, Sha
    Zhang, Shoujiang
    JOURNAL OF ELECTRONIC IMAGING, 2022, 31 (06)
  • [33] Improved RT-DETR Network for High-Quality Defect Detection on Digital Printing Fabric
    Su, Zebin
    Shao, Yunlong
    Li, Pengfei
    Zhang, Xingyi
    Zhang, Huanhuan
    JOURNAL OF NATURAL FIBERS, 2025, 22 (01)
  • [34] Hybrid-DETR: A Differentiated Module-Based Model for Object Detection in Remote Sensing Images
    Yang, Mingji
    Xu, Rongyu
    Yang, Chunyu
    Wu, Haibin
    Wang, Aili
    ELECTRONICS, 2024, 13 (24):
  • [35] Multi-Scale Fusion Uncrewed Aerial Vehicle Detection Based on RT-DETR
    Zhu, Minling
    Kong, En
    ELECTRONICS, 2024, 13 (08)
  • [36] BiF-DETR:Remote sensing object detection based on Bidirectional information fusion
    Xu, Zhijing
    Wang, Chao
    Huang, Kan
    DISPLAYS, 2024, 84
  • [37] Progressive Dynamic Queries Reformation-Based DETR for Remote Sensing Object Detection
    Yin, Haitao
    Wang, He
    Zhu, Zhuyun
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2025, 22
  • [38] Improved object detection method for autonomous driving based on DETR
    Zhao, Huaqi
    Zhang, Songnan
    Peng, Xiang
    Lu, Zhengguang
    Li, Guojing
    FRONTIERS IN NEUROROBOTICS, 2025, 18
  • [39] Utilizing RT-DETR Model for Fruit Calorie Estimation from Digital Images
    Tang, Shaomei
    Yan, Weiqi
    INFORMATION, 2024, 15 (08)
  • [40] Research on Microscale Vehicle Logo Detection Based on Real-Time DEtection TRansformer (RT-DETR)
    Jin, Meiting
    Zhang, Junxing
    SENSORS, 2024, 24 (21)