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 条
  • [1] Drone-DETR: Efficient Small Object Detection for Remote Sensing Image Using Enhanced RT-DETR Model
    Kong, Yaning
    Shang, Xiangfeng
    Jia, Shijie
    SENSORS, 2024, 24 (17)
  • [2] Vehicle detection algorithm based on improved RT-DETR
    Wang, Yuhai
    Xu, Shuobo
    Wang, Peng
    Liu, Lele
    Li, Yanshun
    Song, Ze
    JOURNAL OF SUPERCOMPUTING, 2025, 81 (01):
  • [3] Railway rutting defects detection based on improved RT-DETR
    Yu, Chenghai
    Chen, Xiangwei
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (04)
  • [4] Concrete Bridge Crack Detection Method Based on an Improved RT-DETR Model
    Dai, Lu
    Wang, Di
    Song, Fengquan
    Yang, Han
    2024 3RD INTERNATIONAL CONFERENCE ON ROBOTICS, ARTIFICIAL INTELLIGENCE AND INTELLIGENT CONTROL, RAIIC 2024, 2024, : 172 - 175
  • [5] Improved Road Defect Detection Model Based on RT-DETR for UAV Images
    Wang, Xueqiu
    Gao, Huanbing
    Jia, Zemeng
    Li, Zijian
    Yang, Zhiqiang
    NEURAL COMPUTING FOR ADVANCED APPLICATIONS, NCAA 2024, PT III, 2025, 2183 : 384 - 396
  • [6] RT-DETR-Tomato: Tomato Target Detection Algorithm Based on Improved RT-DETR for Agricultural Safety Production
    Zhao, Zhimin
    Chen, Shuo
    Ge, Yuheng
    Yang, Penghao
    Wang, Yunkun
    Song, Yunsheng
    APPLIED SCIENCES-BASEL, 2024, 14 (14):
  • [7] DV-DETR: Improved UAV Aerial Small Target Detection Algorithm Based on RT-DETR
    Wei, Xiaolong
    Yin, Ling
    Zhang, Liangliang
    Wu, Fei
    SENSORS, 2024, 24 (22)
  • [8] Lightweight tomato ripeness detection algorithm based on the improved RT-DETR
    Wang, Sen
    Jiang, Huiping
    Yang, Jixiang
    Ma, Xuan
    Chen, Jiamin
    Li, Zhongjie
    Tang, Xingqun
    FRONTIERS IN PLANT SCIENCE, 2024, 15
  • [9] LR-DETR: a lightweight real-time traffic sign detection model based on improved RT-DETR
    Zhang, Longzhen
    Wang, Mingyang
    Zhao, Xianhao
    Wang, Xianjie
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2025, 22 (02)
  • [10] UW-DETR: Feature Fusion Enhanced RT-DETR for Improving Underwater Object Detection
    Li, Xingkun
    Wang, Yugang
    Zhao, Yuhao
    Chen, Guodong
    IEEE ACCESS, 2024, 12 : 191967 - 191979