改进RT-DETR的航拍小目标检测算法

作     者：刘思元 高凯 雍龙泉 LIU Siyuan;GAO Kai;YONG Longquan

作者机构：陕西理工大学数学与计算机科学学院陕西汉中723001 

基　　金：陕西省自然科学基础研究计划项目(2024JC-YBMS-014) 陕西省教育厅青年创新团队项目(23JP024) 陕西理工大学科研项目(SLGNL202409) 

出 版 物：《计算机工程与应用》 (Computer Engineering and Applications)

年 卷 期：2025年第61卷第4期

页      码：272-281页

摘      要：针对现有的目标检测算法在航拍图像中的小目标上易出现的漏检和误检问题,提出了基于改进RT-DETR(real-time detection transformer)的算法。在主干网络中引入了部分卷积(partial convolution,PConv),设计了PConvBlock结构,并通过由PConvBlock组成的BasicBlock-PConvBlock模块替代原有BasicBlock,有效减少了模型参数。采用双向特征金字塔网络(bidirectional feature pyramid network,BiFPN)结构优化特征融合模块,并引入S2特征进一步提升小目标的检测能力。引入CARAFE上采样算子,增强了多尺度特征的快速融合。实验表明,在VisDrone测试集上,改进后的模型在参数量上比RT-DETR模型降低了13.9%,同时在mAP0.5和mAP0.5:0.95指标上分别提升了2.4和1.9个百分点。在TT100K和DOTA数据集上均优于RT-DETR算法。改进模型在保持较小参数量和计算量的同时,提高了检测精度,满足了无人机航拍图像实时检测的应用需求。

主 题 词：小目标检测 轻量化 RT-DETR 部分卷积 

学科分类：08[工学] 080203[080203] 0802[工学-机械学] 

核心收录：

D　O　I：10.3778/j.issn.1002-8331.2407-0399

馆 藏 号：203157295...