一种基于注意力掩膜的红外图像彩色化方法

一种基于注意力掩膜的红外图像彩色化方法，红外图像输入初步映射模块生成初步彩色图像，初步彩色图像输入到注意力掩膜模块生成内容掩膜、前景注意力掩膜和背景注意力掩膜；将前景注意力掩膜和背景注意力掩膜与相应的内容掩膜相融合，以增强特定区域的颜色和细节，同时得到目标图像；确定掩膜生成器的循环一致性损失，确保每个输入图像经过循环后生成的图像能够恢复到原始输入图像；确定感知损失，用于提高生成图像的感知质量；输出彩色图像。本发明在完成红外图像彩色化任务的同时显著提高了对于红外图像的前景内容中包含的重要信息的彩色化效果，使得红外图像彩色化后的细节内容得以更加清晰且完整地展现。