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 条
  • [31] ISTD-DETR: A deep learning algorithm based on DETR and Super-resolution for infrared small target detection
    Yang, Huanyu
    Wang, Jun
    Bo, Yuming
    Wang, Jiacun
    NEUROCOMPUTING, 2025, 621
  • [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] Detecting surface defects of pine wood using an improved RT-DETR model
    Hu J.
    Zhang G.
    Shen M.
    Li W.
    Nongye Gongcheng Xuebao/Transactions of the Chinese Society of Agricultural Engineering, 2024, 40 (07): : 210 - 218
  • [34] OHEH-RTDETR: an improved RT-DETR detection model based on frequency layered processing and advanced feature selection
    Wang, Haochun
    Zhang, Yungui
    Wu, Weihang
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2025, 36 (04)
  • [35] Research on Microscale Vehicle Logo Detection Based on Real-Time DEtection TRansformer (RT-DETR)
    Jin, Meiting
    Zhang, Junxing
    SENSORS, 2024, 24 (21)
  • [36] GM-DETR: Research on a Defect Detection Method Based on Improved DETR
    Liu, Xin
    Yang, Xudong
    Shao, Lianhe
    Wang, Xihan
    Gao, Quanli
    Shi, Hongbo
    SENSORS, 2024, 24 (11)
  • [37] Intelligent detection of coal gangue in mining Operations using multispectral imaging and enhanced RT-DETR algorithm for efficient sorting
    Yan, Pengcheng
    Wen, Ziming
    Wu, Zhiqi
    Li, Guodong
    Zhao, Yuting
    Wang, Jingbao
    Wang, Wenchang
    MICROCHEMICAL JOURNAL, 2024, 207
  • [38] 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)
  • [39] DCEI-RTDETR: an improved RT-DETR-based detection algorithm for data center equipment indicator lights
    Xu, Liangjie
    Zeng, Wenxian
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2025, 22 (01)
  • [40] Enhanced real-time detection transformer (RT-DETR) for robotic inspection of underwater bridge pier cracks
    Lv, Zhenming
    Dong, Shaojiang
    Xia, Zongyou
    He, Jingyao
    Zhang, Jiawei
    AUTOMATION IN CONSTRUCTION, 2025, 170