Object detection on visible (RGB) and infrared (IR) images, as an emerging solution to facilitate robust detection for around-the-clock applications, has received extensive attention in recent years. With the help of IR images, object detectors have been more reliable and robust in practical applications by using RGB-IR combined information. However, existing methods still suffer from modality miscalibration and fusion imprecision problems. Since transformer has the powerful capability to model the pairwise correlations between different features, in this paper, we propose a novel Calibrated and Complementary Transformer called $\mathrm{C}^2$Former to address these two problems simultaneously. In $\mathrm{C}^2$Former, we design an Inter-modality Cross-Attention (ICA) module to obtain the calibrated and complementary features by learning the cross-attention relationship between the RGB and IR modality. To reduce the computational cost caused by computing the global attention in ICA, an Adaptive Feature Sampling (AFS) module is introduced to decrease the dimension of feature maps. Because $\mathrm{C}^2$Former performs in the feature domain, it can be embedded into existed RGB-IR object detectors via the backbone network. Thus, one single-stage and one two-stage object detector both incorporating our $\mathrm{C}^2$Former are constructed to evaluate its effectiveness and versatility. With extensive experiments on the DroneVehicle and KAIST RGB-IR datasets, we verify that our method can fully utilize the RGB-IR complementary information and achieve robust detection results. The code is available at https://github.com/yuanmaoxun/Calibrated-and-Complementary-Transformer-for-RGB-Infrared-Object-Detection.git.
翻译:可见光(RGB)与红外(IR)图像目标检测作为实现全天候鲁棒检测的新兴方案,近年来受到广泛关注。借助红外图像的辅助,融合RGB-IR联合信息的目标检测器在实际应用中展现出更高的可靠性与鲁棒性。然而,现有方法仍存在模态校准失准与融合精度不足的问题。鉴于Transformer具备强大的特征间成对关联建模能力,本文提出一种新型校准与互补Transformer——$\mathrm{C}^2$Former,以同步解决上述两个问题。在$\mathrm{C}^2$Former中,我们设计了跨模态交叉注意力(ICA)模块,通过学习RGB与IR模态间的交叉注意力关系,获取校准与互补特征。为降低ICA全局注意力计算带来的计算开销,引入自适应特征采样(AFS)模块以减小特征图维度。由于$\mathrm{C}^2$Former在特征域中执行操作,可通过主干网络嵌入现有RGB-IR目标检测器。据此,我们分别构建了融合$\mathrm{C}^2$Former的单阶段与两阶段目标检测器,以评估其有效性与通用性。在DroneVehicle与KAIST RGB-IR数据集上的大量实验表明,本方法能够充分利用RGB-IR互补信息,实现鲁棒检测结果。代码开源地址:https://github.com/yuanmaoxun/Calibrated-and-Complementary-Transformer-for-RGB-Infrared-Object-Detection.git。