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 条
  • [21] Recognition of Foal Nursing Behavior Based on an Improved RT-DETR Model
    Liu, Yanhong
    Zhou, Fang
    Zheng, Wenxin
    Bai, Tao
    Chen, Xinwen
    Guo, Leifeng
    ANIMALS, 2025, 15 (03):
  • [22] Eye Landmarks Detection using RT-DETR with Rules
    Boonnithititikul, Chatree
    Jaknamon, Teetouch
    Chawuthai, Rathachai
    2024 21ST INTERNATIONAL CONFERENCE ON ELECTRICAL ENGINEERING/ELECTRONICS, COMPUTER, TELECOMMUNICATIONS AND INFORMATION TECHNOLOGY, ECTI-CON 2024, 2024,
  • [23] 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
  • [24] A Comparative Analysis of RT-DETR and YOLOv8 for Urban Zone Aerial Object Detection
    Jun, Eddy Lai Thin
    Tham, Mau-Luen
    Kwan, Ban-Hoe
    2024 IEEE INTERNATIONAL CONFERENCE ON AUTOMATIC CONTROL AND INTELLIGENT SYSTEMS, I2CACIS 2024, 2024, : 340 - 345
  • [25] Research on Rail Defect Recognition Method Based on Improved RT-DETR Model
    Liu, Yue
    Cao, Yuan
    Sun, Yongkui
    2024 5TH INTERNATIONAL CONFERENCE ON COMPUTER ENGINEERING AND APPLICATION, ICCEA 2024, 2024, : 1464 - 1468
  • [26] FedsNet: the real-time network for pedestrian detection based on RT-DETR
    Peng, Hao
    Chen, Shiqiang
    JOURNAL OF REAL-TIME IMAGE PROCESSING, 2024, 21 (04)
  • [27] MSRT-DETR: A novel RT-DETR model with multi-scale feature sequence for cell detection
    Zhou, Chuncheng
    He, Haiyang
    Zhou, Hao
    Ge, Fuhua
    Yu, Pengfei
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2025, 103
  • [28] DETR Novel Small Target Detection Algorithm Based on Swin Transformer
    Xu, Fengchang
    Alfred, Rayner
    Pailus, Rayner Henry
    Lyu, Ge
    Du, Shifeng
    Chew, Jackel Vui Lung
    Li, Guozhang
    Wang, Xinliang
    IEEE ACCESS, 2024, 12 : 115838 - 115852
  • [29] An Efficient Real-time Metal Crack Detection Model Based on RT-DETR
    Cheng, Zhang
    Yang, Fan
    Deng, Shusen
    Chen, Fenglin
    Huo, Junzhou
    2024 9TH ASIA-PACIFIC CONFERENCE ON INTELLIGENT ROBOT SYSTEMS, ACIRS, 2024, : 220 - 224
  • [30] 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)