跨尺度特征引导与交叉注意力融合的多模态目标检测

Cross-Scale Feature Guided and Cross-Attention Fusion Multimodal Object Detection

  • 摘要: 多模态融合目标检测在全天候目标检测领域中具有广泛应用,但当前融合方法主要依赖于简单的累积操作,忽略了模态特征差异对跨模态融合性能的影响。针对上述问题,本文提出一种跨尺度特征引导与交叉注意力融合的多模态目标检测算法(Cross-Scale Feature Guidance and Cross-Attention Fusion for Multimodal Object Detection,CSMF)。使用跨尺度交叉注意力融合模块(Cross-Scale Cross-attention Fusion Module,CCAFM)对可见光和红外图像进行融合,通过跨尺度特征的引导,学习多模态的交叉注意力关系,获得高效融合后的特征。引入感受野增强模块(Receptive Field Enhancement Module,RFE)和自适应快速空间金字塔池化(Adaptive Spatial Pyramid Pooling-Fast,ASPPF),以丰富特征表达。在数据层面上,使用跨模态数据增强方法(Cross-Modal Data Swapping,CDS),有效地捕获局部区域内跨模态相关性。实验结果证明,CSMF算法在FLIR和LLVIP数据集上的检测性能均优于现有的一些主流方法,对比原始算法YOLOv8在mAP50上分别提升了5.3%和1.6%,在复杂环境下检测能力有显著提升。

     

    Abstract: Multimodal fusion target detection has wide applications in all-weather target detection. However, current fusion methods mainly rely on simple accumulation operations and neglect the impact of modal feature differences on cross-modal fusion performance. To address this problem, this study introduces a multimodal target detection algorithm, known as the Cross-Scale Cross-Attention Fusion Module (CSMF), which incorporates cross-scale feature guidance and cross-attention fusion. The CSCM was employed to fuse visible and infrared images. Through cross-scale feature guidance, it learns the cross-attention relationships between the modalities, resulting in efficiently fused features. Moreover, a receptive field enhancement module (RFE) and an adaptive spatial pyramid pooling module (ASPPF) were introduced to enrich feature representations. At the data level, a cross-modal data augmentation method, CDS, was employed to effectively capture cross-modal correlations in local regions. The experimental results revealed that the CSMF algorithm outperformed several existing mainstream methods on the FLIR and LLVIP datasets, improving the mAP50 by 5.3% and 1.6%, respectively, compared with the original algorithms, with a significant enhancement in detection performance within complex environments.

     

/

返回文章
返回