基于循环生成对抗网络的图像着色处理方法及装置

    公开(公告)号:CN111524205A

    公开(公告)日:2020-08-11

    申请号:CN202010325399.6

    申请日:2020-04-23

    Abstract: 本发明公开了一种基于循环生成对抗网络的图像着色处理方法及装置,该方法包括:根据预先获取的图像训练样本,通过机器学习训练一个双层循环生成对抗网络模型,使得模型的生成对抗损失和一致性损失满足预设收敛条件,该模型中包含:用于将灰度图像映射为彩色图像的第一生成器、用于将彩色图像映射为灰度图像的第二生成器、用于区分第一生成器输出彩色图像与真实彩色图像的第一判别器、用于区分第二生成器输出灰度图像与真实灰度图像的第二判别器;在模型的生成对抗损失和一致性损失满足预设收敛条件的情况下,采用第一生成器对待着色的灰度图像进行着色处理。本发明能够避免复杂背景下图像的误着色,提高图像着色处理的准确性。

Patent Agency Ranking