一种基于生成对抗网络的红外图像彩色化方法和系统
摘要:
本发明属于图像处理技术领域,尤其为一种基于生成对抗网络的红外图像彩色化方法,具体步骤为:构建网络模型:构建包括生成器和鉴别器的生成对抗网络;准备数据集:根据第一红外图像数据集对生成对抗网络进行预训练;训练网络模型:利用第一红外图像数据集对网络模型进行训练,直至达到预设阈值;微调模型:利用第二红外图像数据集对网络模型进行再次训练和微调,获得最终模型;保存模型:将获得最终模型的参数进行固化,保存模型。本发明采用的网络结构是生成对抗网络结构,利用生成器和鉴别器之间的博弈特性,加强对图像深层信息的提取,增强彩色化图像的自然度和真实度,动态提高彩色化图像质量。
0/0