一种基于注意力掩膜的红外图像彩色化方法
技术特征：
1.一种基于注意力掩膜的红外图像彩色化方法，其特征在于，包括以下步骤：
2.根据权利要求1所述基于注意力掩膜的红外图像彩色化方法，其特征在于，在步骤1.1中，所述第一激活函数为relu激活函数。
3.根据权利要求1所述基于注意力掩膜的红外图像彩色化方法，其特征在于，在步骤1.2中，所述转换层由9个结构相同的resnet模块组成，resnet模块是五层卷积神经网络。
4.根据权利要求1所述基于注意力掩膜的红外图像彩色化方法，其特征在于，在步骤1.3中，所述第二激活函数为tanh激活函数。
5.根据权利要求1所述基于注意力掩膜的红外图像彩色化方法，其特征在于，在步骤2.2中，所述内容掩膜生成器按照下式生成内容掩膜，
6.根据权利要求5所述基于注意力掩膜的红外图像彩色化方法，其特征在于，在步骤2.3中，所述注意力掩膜生成器按照下式生成前景注意力掩膜和背景注意力掩膜，
7.根据权利要求1所述基于注意力掩膜的红外图像彩色化方法，其特征在于，在步骤3中，所述前景注意力掩膜和背景注意力掩膜与相应的内容掩膜按照下式相融合，得到目标图像，
技术总结
一种基于注意力掩膜的红外图像彩色化方法，红外图像输入初步映射模块生成初步彩色图像，初步彩色图像输入到注意力掩膜模块生成内容掩膜、前景注意力掩膜和背景注意力掩膜；将前景注意力掩膜和背景注意力掩膜与相应的内容掩膜相融合，以增强特定区域的颜色和细节，同时得到目标图像；确定掩膜生成器的循环一致性损失，确保每个输入图像经过循环后生成的图像能够恢复到原始输入图像；确定感知损失，用于提高生成图像的感知质量；输出彩色图像。本发明在完成红外图像彩色化任务的同时显著提高了对于红外图像的前景内容中包含的重要信息的彩色化效果，使得红外图像彩色化后的细节内容得以更加清晰且完整地展现。
技术研发人员：李良福,任文超,温仕羿
受保护的技术使用者：陕西师范大学
技术研发日：
技术公布日：2024/11/18
文档序号 :
【 40048692 】
技术研发人员：李良福,任文超,温仕羿
技术所有人：陕西师范大学
备 注：该技术已申请专利，仅供学习研究，如用于商业用途，请联系技术所有人。
声 明 ：此信息收集于网络，如果你是此专利的发明人不想本网站收录此信息请联系我们，我们会在第一时间删除
技术研发人员：李良福,任文超,温仕羿
技术所有人：陕西师范大学
备 注：该技术已申请专利，仅供学习研究，如用于商业用途，请联系技术所有人。
声 明 ：此信息收集于网络，如果你是此专利的发明人不想本网站收录此信息请联系我们，我们会在第一时间删除