改进RT-DETR的无人机小目标检测算法
DOI:
CSTR:
作者:
作者单位:

1.桂林电子科技大学广西精密导航技术与应用重点实验室 桂林 541004; 2.桂林电子科技大学信息与通信学院 桂林 541004; 3.时空信息与智能位置服务国际合作联合实验室 桂林 541004; 4.广西产研院时空信息技术研究所有限公司 南宁 530023

作者简介:

通讯作者:

中图分类号:

TP391.41;TN911.73

基金项目:

广西科学技术厅项目(AB23026120)、南宁市科学研究与技术开发项目(20231029)、广西科技计划项目(桂科AA23062038,桂科ZY23055048,桂科AB23026120,桂科AA24206025,桂科AA24206043)、国家自然科学基金(U23A20280,62161007,62471153)、产研计划项目(CYY-HT2023-JSJJ-0023-1,CYY-HT2023-JSJJ-0024-1)、广西科技基地和人才专项(桂科AD25069103)资助


Improved UAV small object detection algorithm based on RT-DETR
Author:
Affiliation:

1.Guangxi Key Laboratory of Precision Navigation Technology and Application, Guilin University of Electronic Technology,Guilin 541004, China; 2.School of Information and Communication, Guilin University of Electronic Technology,Guilin 541004, China; 3.Joint Laboratory of International Cooperation on Spatio-Temporal Information and Intelligent Location Services,Guilin 541004, China; 4.Guangxi Institute of Industry and Research Spatio-Temporal Information Technology Co., Ltd.,Nanning 530023, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对无人机视角下目标尺度小、遮挡严重导致的特征提取困难及检测精度偏低问题,提出了一种改进RT-DETR的无人机小目标检测算法SwiftHawk-DETR。首先,重构特征金字塔网络(FPN)的融合路径,新增微小目标检测层P2以强化细粒度特征捕捉,移除冗余大目标检测层P5以精简计算;其次,设计可变形快速多尺度注意网络Dfaster_net,通过动态采样机制强化小目标边界细节捕捉,同时采用高效特征提取策略降低计算开销;然后,在颈部网络中引入小波特征升级(WFU)缓解原模型上采样与Concat操作导致的边缘高频特征失真,结合SlimNeck架构优化跨尺度特征融合;最后,构建Focaler EIoU与NWD加权融合的损失函数,通过平衡定位误差与特征分布差异提升小目标建模能力。实验结果表明,算法在VisDrone2019和HIT-UAV数据集上,mAP50和mAP50:95较基准RT-DETR分别提升3.8%和4.2%,参数量降低61%。

    Abstract:

    To address the problems of difficult feature extraction and low detection accuracy caused by small target scales and severe occlusion from the UAV perspective, this paper proposes an improved RT-DETR-based UAV small target detection algorithm named SwiftHawk-DETR. Firstly, the fusion path of the feature pyramid network (FPN) is reconstructed: A tiny target detection layer P2 is added to enhance fine-grained feature capture, and the redundant large target detection layer P5 is removed to simplify computations. Secondly, a deformable fast multi-scale attention network (Dfaster_net) is designed, which strengthens the capture of small target boundary details through a dynamic sampling mechanism while adopting an efficient feature extraction strategy to reduce computational overhead. Thirdly, wavelet feature upgrade (WFU) is introduced into the neck network to alleviate the distortion of edge high-frequency features caused by upsampling and Concat operations in the original model and the slimneck architecture is combined to optimize cross-scale feature fusion. Finally, a loss function based on the weighted fusion of Focaler EIoU and NWD is constructed to improve the small target modeling ability by balancing localization errors and feature distribution differences.Experimental results on the VisDrone2019 and HIT-UAV datasets show that compared with the baseline RT-DETR, the proposed algorithm increases mAP50 and mAP50:95 by 3.8% and 4.2% respectively, and reduces the number of parameters by 61%.

    参考文献
    相似文献
    引证文献
引用本文

符强,彭章伟,纪元法,任风华.改进RT-DETR的无人机小目标检测算法[J].电子测量技术,2026,49(13):203-215

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-09-08
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告