摘要
本申请实施例提供了一种图像上色方法、图像上色网络的训练方法及装置,该方法包括:获取待上色线稿图和上色参考图,通过图像上色网络对该线稿图执行上色操作:分别提取该线稿图的线稿图特征、上色参考图的参考图特征和颜色语义特征,向线稿图特征中添加最大加噪步数的噪声得到带噪线稿图特征,向参考图特征中分别添加每次去噪所对应的加噪步数的噪声,得到每次去噪对应的带噪参考图特征,基于每次去噪对应的带噪参考图特征和颜色语义特征,对带噪线稿图特征进行迭代去噪得到去噪上色图特征,通过对去噪上色图特征解码得到线稿图对应的上色图。该方法以添加了相应噪声的带噪参考图特征为条件进行引导去噪,提升了去噪效果以及图像上色的准确率。