IF-DETR:改进RT-DETR的航拍图像检测算法
DOI:
CSTR:
作者:
作者单位:

上海电力大学人工智能学部 上海 201306

作者简介:

通讯作者:

中图分类号:

TN911.73; TP391.41

基金项目:

国家自然科学基金(62576203)项目资助


IF-DETR: Improved RT-DETR-based aerial image detection algorithm
Author:
Affiliation:

Aritifical Intelligence Department, Shanghai University of Electricity Power,Shanghai 201306, China

Fund Project:

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

    针对无人机航拍图像中存在的复杂光照干扰,小目标分辨率低、容易漏检误检以及目标结构模糊等问题,提出了一种基于改进RT-DETR的无人机航拍图像目标检测方法IF-DETR。首先,在骨干网络中设计了跨阶段光照门控单元,通过在特征层面建模光照与反射分量,增强了模型在复杂光照环境下对目标的提取能力。其次,在颈部网络构建了小目标多尺度特征融合结构,有效提升了对小目标的特征提取与辨识能力。此外,提出了一种高斯特征增强模块,利用高斯先验与空间注意力机制增强与目标相关的结构信息,同时抑制背景噪声。最后,构建Focaler-Powerful-IoU损失函数,实现了更稳定、更精准的目标定位。实验结果表明,与RT-DETR模型相比,改进后的IF-DETR算法在Visdrone2019数据集上,平均精度均值(mAP)mAP50、mAP50:95、召回率和准确率分别提升了3.0%、2.3%、3.9%、2.0%,能够有效改善无人机航拍图像检测的漏检、误检问题并且提高了检测性能,具有广泛的应用前景。

    Abstract:

    Aiming at the problems in UAV aerial images, such as complex illumination interference, low resolution of small targets with high susceptibility to missing detection and false detection and blurred target structures, a target detection method for UAV aerial images based on improved RT-DETR is proposed, denoted as IF-DETR. Firstly, a cross-stage illumination gating unit is designed in the backbone network, which enhances the model′s ability to extract targets under complex illumination environments by modeling illumination and reflection components at the feature level. Secondly, a multi-scale feature fusion structure for small targets is constructed in the neck network, effectively improving the feature extraction and recognition capabilities for small targets. In addition, a Gaussian feature enhancement module is proposed, which utilizes Gaussian prior and spatial attention mechanism to enhance target-related structural information while suppressing background noise. Finally, a Focaler-Powerful-IoU loss function is constructed to achieve more stable and accurate target localization. Experimental results show that compared with the RT-DETR model, the improved IF-DETR algorithm on the Visdrone2019 dataset achieves improvements of 3.0%, 2.3%, 3.9% and 2.0% in mAP50, mAP50:95, recall and precision respectively. It can effectively alleviate the problems of missing detection and false detection in UAV aerial image detection, improve detection performance, and has broad application prospects.

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

杨洋,魏为民,郑书情,马凯楠,张劲阳. IF-DETR:改进RT-DETR的航拍图像检测算法[J].电子测量技术,2026,49(13):224-234

复制
分享
相关视频

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

重要通知公告

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