一种Restormer结合细节补偿的红外与可见光图像融合方法
doi: 10.11918/202409008
杨艳春 , 李佳龙
兰州交通大学 电子与信息工程学院,兰州 730070
基金项目: 国家自然科学基金(62462043,62067006) ; 甘肃省重点研发计划(25YFGA047) ; 甘肃省自然科学基金(23JRRA847,21JR7RA300)
A Restormer-based fusion method with detail compensation for infrared and visible images
YANG Yanchun , LI Jialong
School of Electronic and Information Engineering, Lanzhou Jiaotong University, Lanzhou 730070 , China
摘要
为提升融合图像的质量和信息完整性,解决红外与可见光图像融合中存在的特征提取能力不足、缺乏纹理细节以及全局上下文信息丢失等问题,提出一种红外与可见光图像的融合与分解网络架构。首先,利用Restormer和Res2Net的并联结构,通过多个深度卷积头转置注意力机制和多尺度残差连接,协同捕获全局上下文信息和局部细节特征;其次,通过带有仿射耦合结构的可逆神经网络,将红外与可见光图像浅层特征分为两部分,利用交替耦合变换实现特征无损保留;然后,重建模块利用拼接及卷积操作生成高质量融合图像;最后,分解网络通过最小化分解损失函数,将融合图像逆向分解为源图像。实验结果表明:在RoadScene数据集上,本文方法的主客观结果均优于多数对比方法,其中标准差、差异相关系数、平均梯度和空间频率较其他对比方法分别平均提升了8.5%、23.1%、49.0%和56.1%;在MSRS数据集上,本文方法较SDCFusion方法在标准差、视觉信息保真度、平均梯度、差异相关系数和空间频率方面分别提升了1.4%、0.4%、0.6%、4.3%和3.4%。所提方法在提升融合图像质量、保留纹理细节和全局信息方面展现出显著优势。
Abstract
To enhance the quality and information integrity of fused images and tackle issues like inadequate feature extraction, insufficient texture details, and loss of global contextual information in infrared and visible images fusion, a fusion and decomposition network architecture for infrared and visible images is proposed. Firstly, a parallel structure of Restormer and Res2Net is utilized. Multiple deep convolutional heads with transposed attention mechanisms and multi-scale residual connections are employed to collaboratively capture the global contextual information and local detail features. Secondly, an invertible neural network with affine coupling structure is adopted to divide the shallow-level features of infrared and visible images into two parts, using alternating coupling transformations to achieve lossless feature preservation. Then, the reconstruction module generates high-quality fused images through concatenation and convolution operations. Finally, the decomposition network reverses the fusion image into source images by minimizing the decomposition loss function. Experimental results show that on the RoadScene dataset, the objective and subjective results of this method surpass most comparative methods. Specifically, compared to other methods, the standard deviation improves by an average of 8.5%, the difference correlation coefficient by 23.1%, the average gradient by 49.0%, and the spatial frequency by 56.1%. On the MSRS dataset, the proposed method outperforms SDCFusion method by 1.4% in standard deviation, 0.4% in visual information fidelity, 0.6% in average gradient, 4.3% in difference correlation coefficient, and 3.4% in spatial frequency. The proposed method shows significant advantages in improving the quality of fused images, preserving texture details, and retaining global information.
由于硬件设备的理论和技术限制,从单一模态传感器或在单一拍摄环境下获得的信息无法有效、全面地描述成像场景。为此,图像融合技术应运而生,其目的是将多模式传感器或在不同拍摄环境下获取的互补信息结合起来[1]。在图像融合技术应用领域,红外与可见光图像融合的应用尤为广泛。可见光图像通过捕捉物体反射的光线,能够呈现出丰富的纹理细节,这与人类视觉习惯相契合,然而这种图像容易受到光照变化等环境因素的干扰,可能会影响其对目标的识别能力。与此同时,红外图像则通过检测物体发出的热辐射来成像,其优势在于高对比度,能够清晰地区分背景与目标,即使在光线不足或环境恶劣的情况下,红外图像依然能够保持较高的成像质量,对目标的识别具有较高的稳定性。通过红外与可见光图像融合,能够得到一种更为全面和鲁棒的图像表示,它不仅能够提供丰富的视觉细节,还能够在各种环境条件下保持对目标的准确识别。这种融合技术在军事侦察、医学成像、遥感监测等多个领域都有着重要的应用价值[2-3],为图像处理领域带来了新的视角和可能性。
现有的红外与可见光图像融合方法主要分为两大类:传统方法和基于深度学习的方法。图像融合的传统技术中,通常采用数学变换来实现源图像的转换和融合。这一过程被划分为特征提取、特征融合和特征重构3个主要步骤。首先,通过特定的数学变换,从源图像中提取关键特征,这是整个特征提取阶段的核心任务。然后,进入特征融合阶段,利用特定的融合规则将提取的特征进行有效整合。最后,在特征重构阶段,对融合后的特征应用逆变换,以重构出最终的融合图像。根据应用数学变换,传统算法可进一步分为4类:基于多尺度分解的方法、基于子空间聚类的方法[4]、基于稀疏表示的方法[5]和基于显著性方法[6]。然而,手工设计的转换过程往往涉及复杂的操作,而融合策略在应对多样化场景时可能会遇到局限。
近年来,随着深度学习的迅速发展,神经网络强大的非线性拟合能力和出色的特征提取能力推动了图像融合领域的发展。目前基于深度学习的融合方法分为基于自动编码器(autoencoder,AE)的方法,基于卷积神经网络(convolutional neural network,CNN)的方法及基于生成对抗网络(generative adversarial network,GAN)的方法。对于AE的融合方法,Li等[7]提出了一种基于密集块的AE,通过密集连接强化细节特征,生成信息丰富的融合图像。考虑到手工融合规则的局限性,Li等[8]提出了一种名为RFN-Nest的残差融合网络,该网络采用巢状连接和注意力机制多尺度捕捉关键信息,确保融合过程的信息完整性。范焱等[9]考虑到频域中潜在的全局信息,提出了一种基于空间和频率特征解耦的融合方法,该方法不仅关注空间域的特征,还结合了频域的信息,通过交互式融合,提高融合算法的鲁棒性。对于CNN的融合方法,李永萍等[10]在变换域中通过多尺度引导滤波器提取基础与细节信息,并利用拉普拉斯能量融合基础层,VGGNet19网络提取细节层特征,最终通过加权平均策略合成融合图像,实现红外与可见光图像的有效融合。Zhang等[11]通过梯度路径和强度路径分别提取图像的高频纹理和像素强度信息,并采用特征重用和信息交换优化融合效果,实现了更好的视觉效果。对于GAN的融合方法,Ma等[12]提出了一种名为FusionGAN的融合方法,该方法首先将GAN引入图像融合领域,可以在没有监督信息的情况下保留重要特征。为避免单鉴别器网络丢失红外图像对比度信息的问题,许光宇等[13]设计了一种双路径双鉴别器生成对抗网络的红外与可见光图像融合方法,该方法通过构建梯度路径和对比度路径,以及多尺度特征提取,提升了融合图像的细节和对比度。利用双鉴别器避免信息丢失,并通过主辅梯度和强度损失函数增强模型的信息提取能力,能够更好地保留和增强图像中的纹理细节。
上述方法虽取得了较好的融合效果,但是目前红外与可见光图像融合多聚焦于从源图像到融合图像的过程,未考虑能否通过分解融合图像来重建源图像,以最小化信息损失;而且融合图像的质量不仅受局部感知区域内像素的影响,还与全局结构的像素强度和纹理细节紧密相关。部分现有方法侧重于局部处理,通过有限的感受野来构建局部深度特征,这种做法虽然能够捕捉到局部细节,但往往忽视了特征之间的长期依赖性,可能导致关键的上下文信息丢失,从而影响融合图像的整体质量和信息的完整性。
为了解决上述问题,本文提出了一种基于Restormer的融合网络和一种基于UNet的分解网络(decomposition network,DN)。融合网络中的特征提取模块采用Restormer与Res2Net并联构成,共同捕捉全局和局部特征,实现信息的有效整合。由于融合图像的纹理细节被认为多数由可见光图像提供,然而一些情况下,红外图像中也可能有少量的细节信息。为了充分挖掘源图像中的细节信息,本文利用可逆神经网络无损信息保存的特性,构建细节补偿模块(detail compensation,DC),使融合图像保留更多边缘和纹理信息。基于分解结果的质量直接取决于融合的结果这一观点,本文基于UNet架构,设计了一个轻量型的分解网络,最大程度地减少信息损失并保留关键的视觉特征。旨在通过本文方法可以优化融合图像中的纹理细节,提升融合图像的视觉效果。
1 相关理论
1.1 Restormer
Parmar等[14]提出了Transformer架构,主要是为了解决机器翻译问题,并在自然语言处理领域取得了显著的成就。近年来,Transformer的思想也被引入到计算机视觉领域。2020年,Dosovitskiy等[15]提出了视觉领域Transformer(vision transformer,ViT),这是一种将Transformer架构应用于图像处理的创新方法,在目标检测、图像分类等视觉任务方面都取得了令人满意的结果。
在处理图像数据时,Transformer通常将图像切割成小块并拉伸成向量处理,这会导致空间信息丢失,影响模型捕捉局部和全局依赖的能力。相比之下,CNN能自然学习局部特征并构建全局特征。此外,Transformer计算量大,结构复杂,包含更多参数和隐藏层,训练和推理成本较高。
为了克服上述局限性,Liu等[16]开发了Swin Transformer,通过移位操作将图像划分为局部和交叉窗口,并在相应窗口内进行注意力计算。然而,该方法将上下文信息的聚合限制在局部邻域内,影响了对长距离依赖关系的捕捉效果。Zamir等[17]提出了Restormer模型,该模型旨在学习长距离依赖关系的同时,保持计算效率。其核心组件包括多个深度卷积头转置注意力(multi-dconv head transposed attention,MDTA)模块和新的门控深度卷积前馈网络(gated dconv feed-forward network,GDFN),如图12所示。MDTA模块有效聚合局部与非局部像素交互,保留细节并捕捉全局信息,使Restormer在处理高分辨率图像时既高效又准确。GDFN模块通过受控特征变换,抑制低信息性特征,仅传递有用信息,增强模型性能,提高泛化能力和鲁棒性。
1多个深度卷积头转置注意力
Fig.1Multi-dconv head transposed attention
2门控深度卷积前馈网络
Fig.2Gated dconv feed-forward network
1.2 可逆神经网络
可逆神经网络(invertible neural network,INN)作为一种特殊的神经网络架构,独具显著特色,即能够实现输入与输出之间的双向、无歧义映射。这种独特的可逆性不仅确保了信息的完整传递,还在反向传播和梯度计算中展现了极高的效率和可靠性。INN的概念最初由Dinh等[18]提出,旨在实现一种特殊的网络结构。其中,输入与输出之间存在明确的可逆关系。给定一个变量y和通过正向计算fθ得到的某种表示或输出,INN能够通过逆向计算过程f-1θ直接且准确地恢复原始的y值。逆向计算的关键在于其反函数的设计,而反函数被构建为与正向计算过程共享参数θ,从而保证了计算的可逆性和效率。
INN的可逆特性为数据重构和恢复提供了强大支持。同时,在图像隐藏、上色、压缩和视频超分辨率等推理任务中也展现了卓越的性能,有效推动了计算机视觉领域的发展。
INN的基本网络架构是采用仿射耦合层来构建的。其核心思想是将输入数据x分为两个部分:u1u2。随后,通过两个学习函数siti对这两部分数据进行转换,并通过交替耦合的方式进行处理,得到两个结果:v1v2,经过一系列变换后得到结果y。函数siti可以是任意复杂度的函数,并且其本身不需要满足可逆性的要求。这种设计为网络提供了更大的灵活性,允许模型学习更复杂的数据转换,同时保持了正向和逆向传播的高效性。正向传播过程的如图3所示,相关的计算式为:
v1=u1exps2u2t2u2
(1)
v2=u2exps1v1t1v1
(2)
3正向传播过程
Fig.3Forward propagation process
INN的逆向过程如图4所示,相关的计算式为:
u2=v2-t1v1exp-s1v1
(3)
u1=v1-t2u2exp-s2u2
(4)
4逆向传播过程
Fig.4Backward propagation process
2 本文方法
2.1 网络总体结构
本文提出的融合与分解网络由融合网络和分解网络构成,如图5所示。其中,融合网络由特征提取模块、细节补偿模块和重建模块组成。融合网络作为图像融合的目标网络,专注于将多源图像融合成一幅单一的、信息丰富的图像。分解网络由双路径UNet组成,致力于从融合结果中分解近似源图像的结果,从而形成一种自我约束的机制,确保融合图像在多个层面上均保持原始图像的信息和细节。整体而言,融合与分解网络架构可以使融合结果包含更丰富的场景细节,从而具有更好的融合效果。
2.2 特征提取模块
特征提取模块由浅层特征提取和深层特征提取两部分组成。浅层特征提取由两个3×3的卷积层组成,而深层特征提取采用Restormer和Res2Net[19]的并联结构,如图6所示。Restormer凭借其卓越的全局上下文信息提取能力,能有效把握图像整体结构;Res2Net则通过在残差块内构建分层的类残差连接,实现多尺度特征的细粒度表示,丰富特征层次并扩大感受野范围,提升图像处理任务表现,尤其在细节纹理信息提取方面表现出色。两者并联协同作用,使深层特征提取模块既能快速捕捉全局上下文信息,又能更好地保留纹理信息,即让模块在关注图像整体特征的同时,深入局部细节,提取出对后续任务至关重要的关键信息。
5网络总体架构
Fig.5Overall network architecture
6深层特征提取
Fig.6Deep feature extraction
浅层与深层特征提取过程表示为:
ϕirs=EsIir
(5)
ϕviss=EsIvis
(6)
ϕird=Edϕirs
(7)
ϕvisd=Edϕviss
(8)
式中:IirIvis分别为一对严格配准的红外和可见光图像,ϕirsϕviss分别为经过浅层特征提取后的红外和可见光特征,ϕirdϕvisd分别为经过深层特征提取后的红外和可见光特征,Es(·)和Ed(·)分别为浅层特征提取操作和深层特征提取操作。
2.3 细节补偿模块
考虑到纹理细节不仅存在于可见光图像中,红外图像也可能包含少量细节,因此将浅层特征提取后的红外与可见光特征拼接在一起,输入至INN中,利用可逆网络的信息无损的这一特点,实现细节信息的有效保留和补偿。
由INN构成的细节补偿模块(DC)如图7所示,通过输入和输出特征相互生成机制,INN模块能够在特征提取过程中最大限度地保留原始输入信息,从而确保细节信息的无损传递。为了进一步增强INN模块的性能,本文设计了带有仿射耦合层的INN块,采用并行映射结构(parallel mapping structure,PMS)(图8)将输入特征分为两部分,分别提取不同方面的特征,通过1×1卷积重新加权和组合特征通道,实现变换操作φ(·)、ρ(·)、η(·)。上述计算过程为:
ϕcat I=Fconcat ϕirs,ϕviss
(9)
ϕcat,k+1I[λ+1:C]=ϕcat,kI[λ+1:C]+φϕcat,kI[1:λ]
(10)
ϕcat,k+1I[1:λ]=ϕcat,kI[1:λ]expρϕcat,k+1I[λ+1:C]))+ηϕcat,k+1I[λ+1:C]
(11)
ϕcat ,k+1I=Fconcat ϕcat ,k+1I[1:λ]+ϕcat ,k+1I[λ+1:C]
(12)
式中:ϕcat I为红外与可见光浅层特征的拼接结果,Fconcat(·)为拼接操作的功能函数,k∈{1,2,3}为细节补偿模块个数,λ为输入特征划分的通道数,C为输入特征的通道总数, φ(·)、ρ(·)、η(·)为INN的变换函数,ϕcatkI [1∶λ]为第k个可逆神经网络中从1到λ的特征,ϕcat k+1I [λ+1∶C]为第k+1个可逆神经网络中从λ+1到C的特征。
7细节补偿模块
Fig.7Detail compensation module
8并行映射结构
Fig.8Parallel mapping structure
2.4 重建模块
重建模块由多个相同的子模块依次堆叠而成,每个子模块包含一个大小为3×3的卷积层和LeakyReLU激活函数。将经过深层特征提取的红外与可见光特征以及补偿信息通过拼接操作一起输入至重建模块中,得到融合图像。该过程表达式为:
ϕF=Fconcat ϕird,ϕvisd,ϕdetail
(13)
Ifu=FRconvϕF
(14)
式中:φdetail为经过细节补偿模块后的特征,φF为红外与可见光图像的深层特征及补偿特征的拼接结果,Ifu为融合结果,FRconv(·)为特征重建操作。
2.5 分解网络
分解网络是对融合后的图像进行分解,以恢复出与原始源图像在细节和结构上高度一致的独立图像。图像分解与融合为互补设计,以确保融合结果的完整性,进而提高图像的质量。本文基于UNet架构设计了轻量型的分解网络,如图5所示。每个分解网络分为下采样层和上采样(Upsample)层两部分。下采样层由3对3×3卷积构成,前两对采用最大池化操作(MaxPooling)作为下采样方法。上采样层也由3对3×3卷积构成,前两对采用双线性插值法作为上采样方法。分解过程表达式为:
IirD=NDFConv3 Ifu
(15)
IvisD=NDFConv 3Ifu
(16)
式中:IirDIvisD分别为经过分解网络分解出的红外与可见光图像,FConv3(·)为对融合图像进行的3×3卷积操作,ND(·)为分解网络对应的操作。
2.6 损失函数
本文网络结构主要分为融合网络和分解网络,相应的损失函数也分为融合损失Lfu和分解损失Lde两部分。总损失Ltotal
Ltotal =Lfu+Lde
(17)
2.6.1 融合损失
融合损失从强度和梯度两方面约束融合结果,表达式为
Lfu=Lint+Lgrad
(18)
式中:Lint为强度损失,Lgrad为梯度损失。
Lint为了突出红外与可见光图像中的显著目标,将结果的强度值最大化,以保证目标的显著性。其表达式为
Lint=1HWIfu-maxIir,Ivis1
(19)
式中:HW分别为图像的高度、宽度,‖·‖1l1范数。
Lgrad保留了两个源图像的最大边缘,以获得更清晰的纹理表示。其表达式为
Lgrad =1HWIfu -maxIir -Ivis 1
(20)
式中为梯度算子,用于测量图像的细粒度纹理信息。
2.6.2 分解损失
分解损失要求融合图像的分解结果尽可能与源图像相似。其表达式为
(21)
式中ij分别为HW的像素索引。
融合图像的质量对于其分解结果与源图像的相似程度具有决定性的影响。因此,分解损失确保融合过程不仅保留了源图像的重要特征,还尽可能地包含了更多的场景内容,进而提升整体的融合性能。
3 实验结果与分析
3.1 数据集和实验设置
本文在MSRS[20]和RoadScene[21]两个公开数据集上进行对比实验。本文方法在MSRS训练集上进行训练,并在其测试集上进行测试以评估其性能。在RoadScene上进行泛化性实验。训练样本随机裁剪成128×128像素的图像块,训练epoch设置为200,训练批次设置为8。
采用Adam优化器进行参数更新,学习率设置为1×10-4,所有实验均在NVIDIA GeForce RTX 4060ti和2.5 GHz Intel Core i5-13400 CPU上采用PyTorch框架进行。
3.2 评价指标和对比方法
为了更好地比较本文方法与其他融合算法的融合性能,选用6种客观评价指标进行评价:标准差DS,从统计学角度衡量融合图像的分布特性及对比度;视觉信息保真度FVI,评价融合图像在人类视觉系统中的信息保真度;平均梯度GA,反映了融合图像中纹理信息的丰富程度;差异相关系数DSC,衡量源图像与融合图像之间的差异,并主要估计从源图像到融合图像传递的信息量;质量指标QAB/F,主要计算从源图像传输到融合图像的边缘信息量;空间频率FS,测量融合图像中存在的空间频率信息。以上6种指标均为值越大,融合效果越好。
将本文方法与目前主流的DenseFuse[7]、RFN-Nest[8]、PMGI[11]、FusionGAN[12]、U2Fusion[21]、IFCNN[22]、SDNet[23]、SeAFusion[24]以及SDCFusion[25]9种深度学习的方法进行比较,以验证其有效性。
3.3 MSRS数据集上的对比实验
3.3.1 主观结果分析
在MSRS测试集中选取5组图像作为主观分析,结果如图9所示。由图9可以看出,DenseFuse在第4组图像的融合结果中,地面纹理不清晰,缺乏细节;第5组图像的融合结果则表现为灯源模糊且亮度偏暗。FusionGAN、SDNet和RFN-Nest的整体融合结果偏暗,导致细节严重丢失,影响了图像的可视性和信息保留。PMGI生成的融合结果具有较高的对比度,但在某些区域出现了局部失真,尤其是在第5组融合结果中,出现了过曝现象。U2Fusion在白天场景中具有理想结果,但在低光场景中表现不佳,红外目标被淹没在黑暗场景中。SeAFusion在第1组和第2组融合结果中,红外目标轮廓清晰,但具体细节模糊。IFCNN和SDCFusion在第3组融合结果中,近景目标清晰,但远景建筑缺少细节,未能有效融合不同距离的目标信息。相比之下,本文提出的方法在5组实验中,无论是在白天还是低光场景,均展现出了卓越的性能,融合结果不仅目标清晰,而且纹理细节丰富,有效地克服了其他方法中存在的一些问题。
3.3.2 客观结果分析
表1为MSRS数据集上20组测试结果的平均值。由表1可知,本文方法在DSFVIGADSCFS中均获得了最优结果。在DSFVIGADSCFS指标方面,本文方法较最新的SDCFusion方法分别提升了1.4%、0.4%、0.6%、4.3%、3.4%。其中,DSFVI说明本文方法融合的图像具有较高的对比度和满意的视觉效果;DSC说明本文方法融合的图像中伪信息较少,源图像与融合图像之间的相关性最大,这归功于本文设计的分解网络;GAFS指标排名第一,说明本文方法融合的结果包含丰富的纹理细节,这得益于本文设计的细节补偿模块;QAB/F仅次于最高的SeAFusion,说明本文方法对源图像边缘信息的融合也有不错的表现。
9MSRS数据集主观实验结果
Fig.9Subjective experimental results on the MSRS dataset
1MSRS数据集20组融合结果平均值
Tab.1Mean of 20 sets of fusion results on the MSRS dataset
为了清晰地呈现各个指标的表现差异,对MSRS数据集中20组融合结果的评价指标进行可视化分析,如图10所示。由图10可以看出,在DSFVIGADSCFS指标中,本文方法在个别指标上虽有轻微波动,但总体趋势优于其他比较方法,表明了本文方法的有效性和可靠性。
10MSRS数据集上20组图像的6种指标分析
Fig.10Plot of the six metrics analyzed for the 20 images on the MSRS dataset
3.4 RoadScene数据集上的对比实验
3.4.1 主观结果分析
在RoadScene数据集中选取5组图像进行主观分析,如图11所示。由图11可以看出:DenseFuse第4组融合结果对于曝光中的物体缺乏细节纹理;FusionGAN、SDNet和RFN-Nest整体融合结果偏暗,物体周围产生伪影;IFCNN第2组融合结果中,地面标识被隐藏在阴影中,模糊不清;PMGI和SDNet的融合结果虽有清晰的红外目标,但缺乏细节,更偏向于红外图像;U2Fusion第3组融合结果中,红外目标的对比度较弱,结果更偏向于可见光图像;SeAFusion第5组融合结果中,门框被完全掩盖于灯光中;SDCFusion第1组和第2组融合结果中,建筑物的边缘出现伪影,树干纹理不清晰。以上对比方法在5组结果中均有不同程度的缺陷,而本文方法在RoadScene数据集中泛化性最优,融合结果保留了源图像的互补信息。
11RoadScene数据集主观实验结果
Fig.11Subjective experimental results on the RoadScene dataset
3.4.2 客观结果分析
表2为RoadScene数据集上20组测试结果的平均值。由表2可以看出,本文方法在DSDSCGAFS中均获得了最优结果。在DSDSCGAFS指标方面,本文方法较其他对比方法分别平均提升了8.5%、23.1%、49.0%和56.1%。说明本文方法的融合图像具有较高的对比度,亮度和对比度分布与源图像保持一致,且在纹理细节和结构相似性方面也表现出色。原因是本文设计的分解网络和细节补偿模块:分解网络能够有效地从融合图像中恢复出源图像,从而约束融合图像质量;细节补偿模块则有效地增强了图像的纹理特征。在FVI指标方面,本文方法仅次于最优的SDCFusion,说明本文方法融合的图像提供了令人满意的视觉效果,在视觉上与人的感受相匹配。在QAB/F指标方面,本文方法仅次于最高的SeAFusion,说明本文方法能够有效地捕捉并融合源图像中的关键边缘特征。
对RoadScene数据集中20组融合结果的评价指标进行可视化分析,如图12所示。由图12可以看出,本文方法在多个指标中处于优势地位,FVIQAB/F的折线虽然略低于SDCFusion和SeAFusion,但仍然保持在较高水平。这些结果证实了本文方法在图像融合领域的有效性和优越性。
2RoadScene数据集20组融合结果平均值
Tab.2Mean of 20 sets of fusion results on the RoadScene dataset
12RoadScene数据集上20组图像的6种指标分析
Fig.12Plot of the six metrics analyzed for the 20 images on the RoadScene dataset
3.5 消融实验
为了验证本文提出的不同模块的有效性,在MSRS数据集进行消融实验。对所提出的并联式深层特征提取模块采用单一模型实验:1)只采用Restormer作为特征提取;2)只采用Res2Net作为特征提取;3)Restormer与Res2Net结合;4)网络仅去除细节补偿模块(DC);5)网络仅去除分解网络(DN);6)本文方法。结果如图13所示。
图13可知,只采用Restormer作为特征提取时,在捕获全局上下文信息方面具有优势,但在局部细节特征的提取方面存在不足,导致地面标识模糊不清。只采用Res2Net作为特征提取时,局部特征和纹理细节的提取表现很好,但在全局信息的整合方面不如Restormer。Restormer与Res2Net结合的方式提供了最佳的全局和局部特征提取能力,人物细节和地面纹理得到了很好地保留。去除细节补偿模块会减少对输入数据细节信息的增强和保留,导致融合图像在纹理细节和边缘清晰度上的损失。没有分解网络,融合图像无法进行有效的逆向分解,导致目标细节丢失,且存在轻微过曝。本文方法结合了Restormer和Res2Net的特征提取能力,细节补偿模块及分解网络的完整网络架构,在图像融合质量、纹理细节的保留及信息的完整性方面提供最佳的表现。
13消融实验结果
Fig.13Results of ablation experiments
表3为消融实验定量分析结果。由表3可以看出,在本文方法中,信息熵EN和结构相似性SS均取得最优,说明整合各个模块显著改善了融合图像的质量。
3消融实验定量分析
Tab.3Quantitative analysis of ablation experiments
3.6 运行效率对比分析
表4为不同对比方法在MSRS和RoadScene数据集上的平均运行时间(tMSRStRoadScene)。由表4可以看出,本文方法的平均运行时间位居第四。DenseFuse、IFCNN和RFN-Nest的平均运行时间均比本文方法低,原因是这3种方法均采用了包含多个卷积层的传统网络结构,并设计了一种加法融合策略。
4不同方法的平均运行时间
Tab.4Average running time of different methods
4 结论
本文提出了一种融合与分解网络架构,并在公开的MSRS和RoadScene数据集上进行实验验证,得出以下结论:
1)提出的深层特征提取模块通过结合Restormer和Res2Net的优势,有效地捕捉了全局和局部特征,增强了网络的表征能力。
2)利用可逆神经网络的特性,本文设计的细节补偿模块显著提升了融合图像的纹理细节和视觉效果,确保了融合结果在保留边缘信息和纹理细节方面的优越性。 3)设计的分解网络能够将融合图像有效分解回原始的红外和可见光图像,确保了融合图像在多个层面上均保持原始图像的信息和细节。
3)RoadScene数据集上的泛化实验得出,在DSDSCGAFS指标方面,本文方法较其他对比方法分别平均提升了8.5%、23.1%、49.0%和56.1%。MSRS数据集上的测试实验得出,在DSFVIGADSCFS指标方面,本文方法较最新的SDCFusion方法分别提升了1.4%、0.4%、0.6%、4.3%和3.4%。
4)本文方法不仅在融合结果上表现出色,而且在运行效率上也具有一定的优势。
1多个深度卷积头转置注意力
Fig.1Multi-dconv head transposed attention
2门控深度卷积前馈网络
Fig.2Gated dconv feed-forward network
3正向传播过程
Fig.3Forward propagation process
4逆向传播过程
Fig.4Backward propagation process
5网络总体架构
Fig.5Overall network architecture
6深层特征提取
Fig.6Deep feature extraction
7细节补偿模块
Fig.7Detail compensation module
8并行映射结构
Fig.8Parallel mapping structure
9MSRS数据集主观实验结果
Fig.9Subjective experimental results on the MSRS dataset
10MSRS数据集上20组图像的6种指标分析
Fig.10Plot of the six metrics analyzed for the 20 images on the MSRS dataset
11RoadScene数据集主观实验结果
Fig.11Subjective experimental results on the RoadScene dataset
12RoadScene数据集上20组图像的6种指标分析
Fig.12Plot of the six metrics analyzed for the 20 images on the RoadScene dataset
13消融实验结果
Fig.13Results of ablation experiments
1MSRS数据集20组融合结果平均值
Tab.1Mean of 20 sets of fusion results on the MSRS dataset
2RoadScene数据集20组融合结果平均值
Tab.2Mean of 20 sets of fusion results on the RoadScene dataset
3消融实验定量分析
Tab.3Quantitative analysis of ablation experiments
4不同方法的平均运行时间
Tab.4Average running time of different methods
KARIM S, TONG Geng, LI Jinyang,et al. Current advances and future perspectives of image fusion:a comprehensive review[J]. Information Fusion,2023,90:185. DOI:10.1016/j.inffus.2022.09.019
唐霖峰, 张浩, 徐涵, 等. 基于深度学习的图像融合方法综述[J]. 中国图象图形学报,2023,28(1):3.TANG Linfeng, ZHANG Hao, XU Han,et al. Deep learning-based image fusion:a survey[J]. Journal of Image and Graphics,2023,28(1):3. DOI:10.11834/jig.220422
张洲宇, 曹云峰, 丁萌. 采用多层卷积稀疏表示的红外与可见光图像融合[J]. 哈尔滨工业大学学报,2021,53(12):51.ZHANG Zhouyu, CAO Yunfeng, DING Meng. Infrared and visible image fusion via multi-layer convolutional sparse representation[J]. Journal of Harbin Institute of Technology,2021,53(12):51. DOI:10.11918/202005038
WANG Haozhe, SHU Chang, LI Xiaofeng,et al. Two-stream edge-aware network for infrared and visible image fusion with multi-level wavelet decomposition[J]. IEEE Access,2024,12:22190. DOI:10.1109/ACCESS.2024.3364050
WU Peicong. Infrared and visible image fusion based on potential low-rank decomposition and anisotropic guided filtering[C]//6th International Conference on Electronics Technology(ICET). Chengdu: IEEE,2023:36. DOI:10.1109/ICET58434.2023.10211821
WANG Yang, CAO Xiaoqian, LIU Weifeng,et al. Improved sparse representation fusion rules based infrared and visible image fusion algorithm[C]//12th IEEE International Conference on Control, Automation and Information Sciences(ICCAIS). Hanoi: IEEE,2023:236. DOI:10.1109/ICCAIS59597.2023.10382352
LI Hui, WU Xiaojun. DenseFuse:a fusion approach to infrared and visible images[J]. IEEE Transactions on Image Processing,2018,28(5):2614. DOI:10.1109/TIP.2018.2887342
LI Hui, WU Xiaojun, KITTLER J. RFN-Nest:an end-to-end residual fusion network for infrared and visible images[J]. Information Fusion,2021,73:72. DOI:10.1016/j.inffus.2021.02.023
范焱, 刘乔, 袁笛, 等. 空域和频域特征解耦的红外与可见光图像融合[J]. 红外与激光工程,2024,53(8):222.FAN Yan, LIU Qiao, YUAN Di,et al. Spatial and frequency domain feature decoupling for infrared and visible image fusion[J]. Infrared and Laser Engineering,2024,53(8):222. DOI:10.3788/IRLA20240198
李永萍, 杨艳春, 党建武, 等. 基于变换域VGGNet19的红外与可见光图像融合[J]. 红外技术,2022,44(12):1293.LI Yongping, YANG Yanchun, DANG Jianwu,et al. Infrared and visible image fusion based on transform domain VGGNet19[J]. Infrared Technology,2022,44(12):1293
ZHANG Hao, XU Han, XIAO Yang,et al. Rethinking the image fusion:a fast unified image fusion network based on proportional maintenance of gradient and intensity[C]//34th AAAI Conference on Artificial Intelligence. New York: AAAI,2020:12797. DOI:10.1609/AAAI. V34i07.6975
MA Jiayi, YU Wei, LIANG Pengwei,et al. FusionGAN:a generative adversarial network for infrared and visible image fusion[J]. Information Fusion,2019,48:11. DOI:10.1016/j.inffus.2018.09.004
许光宇, 陈浩宇, 张杰. 双路径双鉴别器生成对抗网络的红外与可见光图像融合[J]. 计算机辅助设计与图形学学报,2024,36(12):1946.XU Guangyu, CHEN Haoyu, ZHANG Jie. Infrared and visible image fusion based on dual-path and dual-discriminator generation adversarial network[J]. Journal of Computer-Aided Design & Computer Graphics,2024,36(12):1946. DOI:10.3724/SP. J.1089.2024.20170
PARMAR N, VASWANI A, USZKOREIT J,et al. Image transformer[C]//35th International Conference on Machine Learning. Stockholm: IMLS,2018:6453
DOSOVITSKIY A, BEYER L, KOLESNIKOV A,et al. An image is worth 16×16 words:transformers for image recognition at scale[C]//9th International Conference on Learning Representations. Vienna: ICLR,2021
LIU Ze, LIN Yutong, CAO Yue,et al. Swin transformer:hierarchical vision transformer using shifted windows[C]//Proceedings of the IEEE/CVF International Conference on Computer Vision. Montreal: IEEE,2021:10012. DOI:10.1109/ICCV48922.2021.00986
ZAMIR S W, ARORA A, KHAN S,et al. Restormer:efficient transformer for high-resolution image restoration[C]//IEEE/CVF Conference on Computer Vision and Pattern Recognition. New Orlean: IEEE,2022:5728. DOI:10.1109/CVPR52688.2022.00564
DINH L, SOHL-DICKSTEIN J, BENGIO S. Density estimation using real NVP[C]//5th International Conference on Learning Representations. Toulon: ICLR,2017
GAO Shanghua, CHENG Mingming, ZHAO Kai,et al. Res2Net:a new multi-scale backbone architecture[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence,2019,43(2):652. DOI:10.1109/TPAMI.2019.2938758
TANG Linfeng, YUAN Jiteng, ZHANG Hao,et al. PIAFusion:a progressive infrared and visible image fusion network based on illumination aware[J]. Information Fusion,2022,83:79. DOI:10.1016/j.inffus.2022.03.007
XU Han, MA Jiayi, JIANG Junjun,et al. U2Fusion:a unified unsupervised image fusion network[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence,2022,44(1):502. DOI:10.1109/TPAMI.2020.3012548
ZHANG Yu, LIU Yu, SUN Peng,et al. IFCNN:a general image fusion framework based on convolutional neural network[J]. Information Fusion,2020,54:99. DOI:10.1016/j.inffus.2019.07.011
ZHANG Hao, MA Jiayi. SDNet:a versatile squeeze and decomposition network for real-time image fusion[J]. International Journal of Computer Vision,2021,129(10):2761. DOI:10.1007/s11263-021-01501-8
TANG Linfeng, YUAN Jiteng, MA Jiayi. Image fusion in the loop of high-level vision tasks:a semantic-aware real-time infrared and visible image fusion network[J]. Information Fusion,2022,82:28. DOI:10.1016/j.inffus.2021.12.004
LIU Xiaowen, HUO Hongtao, LI Jing,et al. A semantic-driven coupled network for infrared and visible image fusion[J]. Information Fusion,2024,108:102352. DOI:10.1016/j.inffus.2024.102352