一种基于生成对抗式网络的图像着色方法及装置

    公开(公告)号:CN114898031A

    公开(公告)日:2022-08-12

    申请号:CN202210367877.9

    申请日:2022-04-08

    Abstract: 本发明公开了一种基于生成对抗式网络的图像着色方法及装置,该方法包括:获取待着色图像;通过图像分类网络对待着色图像进行分类;根据分类结果,将待着色图像输入对应类别的生成对抗式网络中,对待着色图像中的每个像素值进行预测,输出每个像素的RGB值,最终生成一副彩色图像;判断生成的彩色图像是否满足要求;若满足要求,则将其作为最终的着色效果;否则,从网络上下载与待着色图像的结构和内容相似的图像组成新的数据集,利用新的数据集重新训练生成对抗式网络,利用重新训练后的生成对抗式网络重新生成待着色图像对应的彩色图像。本发明有效改善了颜色错误,数据不足等问题。

Patent Agency Ranking