基于RT-DETR的无人机航拍图像小目标检测算法
DOI:
CSTR:
作者:
作者单位:

贵州大学大数据与信息工程学院 贵阳 550025

作者简介:

通讯作者:

中图分类号:

TP391.4;TN919.8

基金项目:


Small object detection algorithm in Drone aerial images based on RT-DETR
Author:
Affiliation:

College of Big Data and Information Engineering, Guizhou University, Guiyang 550025, China

Fund Project:

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

    随着无人机应用场景不断拓展,航拍图像中小目标检测成为计算机视觉领域的研究热点。针对小目标特征不明显、背景复杂导致误检和漏检,现有算法检测精度与实时性难以兼顾等问题,本研究提出了一种基于RT-DETR的航拍图像小目标检测算法FST-RTDETR来解决这些问题。首先,将FasterNet与EMA注意力机制结合,重新设计原有模块的Basic Block模块的结构,实现提高网络运行速度和视觉任务的准确性。其次,为了解决传统P2检测层添加后出现计算量过大、后处理更加耗时等问题,本研究基于原本的CCFM架构上提出使用P2特征层经过SPDConv得到富含小目标信息的特征给到P3进行融合,然后使用CSP思想和基于Omni-Kernel进行改进得到CSP-OmniKernel进行特征整合,有效地学习从全局到局部的特征表现,最终减少漏检率、误检率和提高小目标的检测性能。最后,为了使得模型简化损失函数计算过程、改进回归效率和精度以及拥有更全面的损失考虑,使用inner-MPDIoU替换原来的GIoU。改进后的算法在VisDrone2019数据集上的实验表明,FST-RTDETR模型实现了49.6%的mAP@50,相对于原来的RT-DETR模型提高了2.1%。FST-RTDETR模型显著提升了无人机图像的目标检测性能,提高了模型效率,对比其他算法表现出了良好的性能。

    Abstract:

    With the continuous expansion of drone application scenarios, small object detection in aerial images has become a research hotspot in the field of computer vision. In view of the problems that small object features are not obvious, complex backgrounds lead to false detection and missed detection, and the existing algorithms are difficult to balance detection accuracy and real-time performance, this paper proposes an aerial image small object detection algorithm FST-RTDETR based on RT-DETR to solve these problems. First, FasterNet is combined with the EMA attention mechanism, and the structure of the Basic Block module of the original module is redesigned to improve the network operation speed and the accuracy of visual tasks. Secondly, in order to solve the problems of excessive calculation and more time-consuming post-processing after adding the traditional P2 detection layer, this study propose to use the P2 feature layer based on the original CCFM architecture to obtain features rich in small object information through SPDConv and give them to P3 for fusion, and then use the CSP idea and Omni-Kernel to improve CSP-OmniKernel for feature integration, effectively learn the feature performance from global to local, and finally reduce the missed detection rate, false detection rate and improve the detection performance of small objects. Finally, in order to simplify the loss function calculation process, improve regression efficiency and accuracy, and have a more comprehensive loss consideration, this study use inner-MPDIoU to replace the original GIoU. Experiments on the improved algorithm on the VisDrone2019 dataset show that the FST-RTDETR model achieves a detection accuracy of 49.6%, which is 2.1% higher than the original RT-DETR model. The FST-RTDETR model significantly improves the object detection performance of drone images, improves model efficiency, and shows good performance compared to other algorithms.

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

刘杰,李志文,张腾庆,谢明山.基于RT-DETR的无人机航拍图像小目标检测算法[J].电子测量技术,2026,49(6):98-109

复制
分享
相关视频

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

重要通知公告

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