DV-DETR: Improved UAV Aerial Small Target Detection Algorithm Based on RT-DETR

被引:0
|
作者
Wei, Xiaolong [1 ]
Yin, Ling [1 ]
Zhang, Liangliang [1 ]
Wu, Fei [1 ]
机构
[1] Shanghai Univ Engn Sci, Sch Elect & Elect Engn, Shanghai 201620, Peoples R China
关键词
transformer; small target detection; real-time task; RT-DETR algorithm;
D O I
10.3390/s24227376
中图分类号
O65 [分析化学];
学科分类号
070302 ; 081704 ;
摘要
For drone-based detection tasks, accurately identifying small-scale targets like people, bicycles, and pedestrians remains a key challenge. In this paper, we propose DV-DETR, an improved detection model based on the Real-Time Detection Transformer (RT-DETR), specifically optimized for small target detection in high-density scenes. To achieve this, we introduce three main enhancements: (1) ResNet18 as the backbone network to improve feature extraction and reduce model complexity; (2) the integration of recalibration attention units and deformable attention mechanisms in the neck network to enhance multi-scale feature fusion and improve localization accuracy; and (3) the use of the Focaler-IoU loss function to better handle the imbalanced distribution of target scales and focus on challenging samples. Experimental results on the VisDrone2019 dataset show that DV-DETR achieves an mAP@0.5 of 50.1%, a 1.7% improvement over the baseline model, while increasing detection speed from 75 FPS to 90 FPS, meeting real-time processing requirements. These improvements not only enhance the model's accuracy and efficiency but also provide practical significance in complex, high-density urban environments, supporting real-world applications in UAV-based surveillance and monitoring tasks.
引用
收藏
页数:15
相关论文
共 50 条
  • [1] Small Target Detection Algorithm for Traffic Signs Based on Improved RT-DETR
    Liang, Nuanling
    Liu, Weisheng
    ENGINEERING LETTERS, 2025, 33 (01) : 140 - 147
  • [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] 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):
  • [4] 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
  • [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] Steel surface defect detection based on the lightweight improved RT-DETR algorithm
    Mao, Haojie
    Gong, Yongwang
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2025, 22 (01)
  • [7] Railway rutting defects detection based on improved RT-DETR
    Yu, Chenghai
    Chen, Xiangwei
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (04)
  • [8] RS-DETR: An Improved Remote Sensing Object Detection Model Based on RT-DETR
    Zhang, Hao
    Ma, Zheng
    Li, Xiang
    APPLIED SCIENCES-BASEL, 2024, 14 (22):
  • [9] FECI-RTDETR a Lightweight Unmanned Aerial Vehicle Infrared Small Target Detector Algorithm Based on RT-DETR
    Xue, Renzheng
    Hua, Shijie
    Xu, Haiqiang
    IEEE ACCESS, 2025, 13 : 9578 - 9591
  • [10] Object Detection in UAV Images Based on RT-DETR with CG Downsampling and CCFMP
    Yu, Chushi
    Shin, Yoan
    2024 IEEE VTS ASIA PACIFIC WIRELESS COMMUNICATIONS SYMPOSIUM, APWCS 2024, 2024,