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

    Aerial small object detection algorithm based on improved RT-DETR

    • 针对航拍图像中目标尺度小、特征信息稀疏及复杂背景干扰导致的检测精度偏低问题,提出了一种基于改进型RT-DETR的目标检测算法。首先,构建双路径增强特征提取网络,融合全维动态卷积与双路径卷积结构,在降低模型参数量的同时强化对微小目标的浅层响应能力。其次,设计轻量高效的特征融合模块,通过引入空间深度转换卷积,将跨阶段部分结构(Cross Stage Partial Network, CSP)与全维卷积机制(OmniKernel)机制有机结合,增强多尺度语义特征的耦合度,从而保留高分辨率空间细节,提升模型在强杂波背景下的目标判别能力。最后,在边界框回归阶段引入FMPDIoU损失函数,以缓解小目标梯度稀疏及正负样本失衡问题。在VisDrone 2019数据集上的实验结果表明:该模型在平均精度均值(mAP)、召回率与准确率上均显著优于多种主流对比算法;相较于基线RT-DETR-R18模型,mAP提升2.1%,小目标平均精度提高1.1%,验证了所提改进策略的有效性与良好的泛化迁移能力。

       

      Abstract: To address the issues of low detection accuracy, insufficient feature extraction, and strong background interference in aerial imagery, this paper proposes an improved RT-DETR object detection algorithm to enhance model performance in complex scenarios. A dual-path enhanced feature extraction module is constructed, introducing a dual-path convolution structure and an omni-dimensional dynamic convolution layer to reduce computational cost while improving the perception of small-scale objects. A lightweight and efficient feature fusion module is then designed, which integrates spatial-depth transform convolution and combines the CSP structure with the OmniKernel mechanism to enhance the coupling of multi-scale semantic information, preserve high-resolution features, and improve the model's ability to distinguish small objects under complex backgrounds. Finally, a FMPDIoU loss function is introduced in the bounding box regression process to alleviate the issues of gradient sparsity and class imbalance in small object detection. Experiments conducted on the VisDrone 2019 dataset show that the proposed model outperforms mainstream object detection algorithms in terms of mAP, recall and precision. Compared with the baseline RT-DETR-R18 model, it achieves a 2.1% improvement in mAP and a 1.1% increase in average precision for small objects, demonstrating excellent detection performance and strong transferability.

       

    /

    返回文章
    返回