基于自注意力机制的红外和可见光图像融合方法
发布时间: 2022-06-30
基本信息
基于自注意力机制的红外和可见光图像融合方法,属于图像融合领域。主要解决图像融合时忽略人眼主观感受的重要性问题。其实现步骤是:1)搭建基于自注意力机制的深度自编码网络结构,以端到端的方式对特征提取,融合规则和重构规则同时学习;2)将编码层不同分支的特征图输入自注意力机制获得注意力图,并采用均值融合策略获得输出特征图;3)设计内容损失和细节损失两类损失函数,分别用于突出红外目标信息,锐化边缘和更好地利用源图像中的纹理细节;4)训练神经网络,并对自注意力机制进行可视化以调整网络结构和损失函数。本发明能通过学习注意力图以最佳方式分配注意力,获取图像关键信息,改善视觉效果,提高融合图像的质量。