• 基于RTDETR的无人机视角目标检测算法

    RTDETR-based algorithm for target detection in drone aerial imagery

    • 为解决无人机视角图像的小目标检测中存在的误检、漏检和检测性能差等问题,提出了一种改进RTDETR模型。在主干网络融入由轻量级可变卷积核 (AKConv)和中值增强空间通道注意力机制组成BasicAK MECS模块,增强了特征提取效果;Neck部分引入Slim-neck,轻量化模型的同时提高检测的准确性和效率;在上采样部分,利用动态尺度序列特征融合(DySSFF)模块,提升小目标特征信息提取能力。损失函数使用Inner-SIoU,注入边界框的形状和尺度信息,增强边界框回归效果。实验结果表明,所提出的模型在 Visdrone2019数据集上的mAP和mAPs分别提升了2.2%和0.9%,有效提高了模型目标检测能力。

       

      Abstract: To address issues such as false positives, missed detections, and poor detection performance in small object detection from drone imagery, an improved RTDETR model is proposed. The backbone network is enhanced with a BasicAK MECS module, which integrates a lightweight adaptable kernel convolution (AKConv) and a median-enhanced spatial channel attention mechanism, thereby improving feature extraction. The Neck component incorporates Slim-neck to both lighten the model and increase detection accuracy and efficiency. In the upsampling stage, a Dynamic Scale Sequence Feature Fusion (DySSFF) module is utilized to enhance the extraction of small object features. The loss function employed is Inner-SIoU, which incorporates shape and scale information of bounding boxes to improve bounding box regression performance. Experimental results demonstrate that the proposed model achieves a 2.2% increase in mAP and a 0.9% increase in mAPs on the Visdrone2019 dataset, significantly enhancing the model's object detection capability.

       

    /

    返回文章
    返回