一种基于人眼视觉特征的自适应对比学习红外图像彩色化方法及系统

本发明公开了一种基于人眼视觉特征的自适应对比学习红外图像彩色化方法及系统，属于图像彩色化技术领域。一种基于人眼视觉特征的自适应对比学习红外图像彩色化方法，包括如下步骤：步骤1：准备红外数据集：对训练集一与训练集二进行预处理，经过预处理固定尺寸训练集图像；训练集一为KAI ST数据集，训练集二为FL IR数据集，通过将原数据集图像作为输入，输出固定尺寸训练集图像；步骤2：构建网络模型：整体网络由一个对抗网络组成，包括生成器、鉴别器、样本标记操作和生成拟合操作组成。本发明引入对比学习，通过构建全新正负样本，并设计生成器，使输出红外图像彩色化结果更加真实，符合人眼视觉观察的局部和整体感受。