基于Swin Transformer的图像融合方法
摘要:
本发明公开了一种基于Swin Transformer的图像融合方法,涉及图像处理技术领域。本发明包括以下步骤:步骤S1、构建端对端神经网络,所述端对端神经网络包括生成器和判别器;步骤S2、构建生成器的损失函数以及判别器中鉴别器的损失函数;步骤S3、利用KAIST数据集的部分原始红外图像和原始可见光图像对端对端神经网络进行训练,得到端对端神经网络模型;步骤S4、测试步骤S3保存的端对端神经网络模型,输出最终融合图像。本发明所述图像融合方法能够解决现有基于自注意力Transformer图像融合方法无法有效提取多尺度特征,无法有效降低计算复杂度以及无法利用先验知识等问题。
公开/授权文献
0/0