《计算机工程与应用杂志》发表论文赏析
作者:向毅伟, 蒋瑜, 王琪凯, 罗熔熔
单位:成都信息工程大学 软件工程学院,成都 610225
摘要:针对无人机目标检测场景中的目标尺度小、遮挡严重、样本分布不均匀等问题,提出了一种改进实时检测Transformer(real-time detection Transformer,RT-DETR)的MSM-DETR检测器。在颈部网络中设计并引入了DSSF特征融合结构,通过结合维度感知选择性整合(DASI)模块以及尺度序列特征融合(SSFF)模块丰富特征融合阶段小目标信息,提高检测精度。针对遮挡严重和样本分布不均匀的问题,提出多核并行尺度间融合(multi-core parallel scale fusion,MCPSF)模块,通过利用尺度间融合思想改进多核分组卷积带来的尺度间信息不平衡问题,为模型提供多尺度感受野,同时利用EMA注意力进一步增强组内上下文信息,提升检测精度。将Inner思想融入原损失函数中,通过引入不同尺度的辅助边框计算损失,加速收敛。实验结果表明,改进后的模型在VisDrone2019数据集中的验证集和测试集的mAP为49.5%、38.9%,较原模型分别提升2.5、2.4个百分点。
关键词:RT-DETR,航拍图像,多尺度,感受野