###
计算机系统应用英文版:2026,35(6):195-210
本文二维码信息
码上扫一扫!
RT-DETR在无人机目标检测中的轻量化与精度增强
(1.南京信息工程大学 软件学院, 南京 210044;2.南京信息工程大学 科技产业处, 南京 210044;3.南京信息工程大学 计算机学院、网络空间安全学院, 南京 210044)
Lightweighting and Precision Enhancement of RT-DETR for UAV Object Detection
(1.School of Software, Nanjing University of Information Science & Technology, Nanjing 210044, China;2.Science and Technology Industry Division, Nanjing University of Information Science & Technology, Nanjing 210044, China;3.School of Computer Science & School of Cyber Science and Engineering, Nanjing University of Information Science & Technology, Nanjing 210044, China)
摘要
图/表
参考文献
相似文献
本文已被:浏览 64次   下载 91
Received:November 04, 2025    Revised:November 27, 2025
中文摘要: 随着无人机技术的快速发展, 基于无人机的目标检测在智慧城市交通监控等领域应用前景广阔; 然而, 无人机图像存在目标尺度变化大、背景复杂等挑战, 传统检测算法难以在精度和效率间取得平衡. 本文提出了一种基于 RT-DETR 的无人机目标检测优化方法. 针对无人机图像中目标尺度变化大的问题, 优化了多尺度特征融合策略, 通过改进 P3–P5 特征金字塔结构, 增强了算法对不同尺度目标的检测能力; 同时, 针对无人机场景中目标通常较小且密集分布的特点, 优化了AIFI 注意力机制, 提高了算法对小目标的特征表达能力; 基于 RT-DETR-R18 轻量化设计, 在保证检测精度的前提下显著降低了计算复杂度. 在包含10个类别无人机图像的VisDrone 数据集上的实验结果表明, 本文提出的方法相比原始 RT-DETR 算法, 在保持实时性的同时提升了小目标检测精度, 验证了各个改进模块的有效性.
Abstract:As unmanned aerial vehicle (UAV) technology develops rapidly, UAV-based object detection has broad application prospects in fields such as smart city traffic monitoring. However, UAV-captured images face challenges such as large object scale variations and complex backgrounds, making it difficult for traditional detection algorithms to balance precision and efficiency. This study proposes an optimized UAV object detection method based on RT-DETR. To address the problem of large target scale variations in UAV-captured images, this study optimizes the multi-scale feature fusion strategy, and the algorithm’s ability to detect objects at different scales is enhanced by improving the P3–P5 feature pyramid structure. Additionally, considering that objects in UAV scenarios are typically small and densely distributed, the AIFI attention mechanism is optimized to improve the algorithm’s feature representation capability for small objects. Based on the lightweight RT-DETR-R18 design, the proposed method significantly reduces computational complexity while maintaining detection precision. The experimental results on the VisDrone dataset which contains UAV-captured images across ten categories demonstrate that compared to the original RT-DETR algorithm, the proposed method improves detection precision for small objects while maintaining real-time performance, validating the effectiveness of each improved module.
文章编号:     中图分类号:    文献标志码:
基金项目:林芝地区交通气象监测预警服务平台项目 (2024-YZ-01); 国家自然科学基金 (41975183)
引用文本:
王军,陆姚沈.RT-DETR在无人机目标检测中的轻量化与精度增强.计算机系统应用,2026,35(6):195-210
WANG Jun,LU Yao-Shen.Lightweighting and Precision Enhancement of RT-DETR for UAV Object Detection.COMPUTER SYSTEMS APPLICATIONS,2026,35(6):195-210