基于预训练固定参数和深度特征调制的红外与可见光图像融合网络

作者：徐少平^*; 周常飞; 肖建; 陶武勇; 戴田宇

来源：电子与信息学报, 2024, 1-9.

• 红外与可见光图像
• 图像融合
• 自监督预训练
• 无监督融合训练
• 固定参数
• 深度特征调制

为了更好地利用红外与可见光图像中互补的图像信息，得到符合人眼感知特性的融合图像，该文采用两阶段训练策略提出一种基于预训练固定参数和深度特征调制的红外与可见光图像融合网络(PDNet)。具体地，在自监督预训练阶段，以大量清晰的自然图像分别作为U型网络结构(UNet)的输入和输出，采用自编码器技术完成预训练。所获得编码器模块能有效提取输入图像的多尺度深度特征功能，而解码器模块则能将其重构为与输入图像差异极小的输出图像；在无监督融合训练阶段，将预训练编码器和解码器模块的网络参数保持固定不变，而在两者之间新增包含Transformer结构的融合模块。其中，Transformer结构中的多头自注意力机制能对编码器分别从红外和可见光图像提取到的深度特征权重进行合理分配，从而在多个尺度上将两者融合调制到自然图像深度特征的流型空间上来，进而保证融合特征经解码器重构后所获得融合图像的视觉感知效果。大量实验表明：与当前主流的融合模型(算法)相比，所提PDNet模型在多个客观评价指标方面具有显著优势，而在主观视觉评价上，也更符合人眼视觉感知特点。