基于权重学习的着色方法

为实现灰度图像的着色,提出一种基于相似关系的着色方法,同时需要用户对图像做颜色标记。权重代表像素之间的一种相似关系,决定了颜色传递的能量大小。为了获得准确的权重信息,提出了权重学习的方法,并建立了从灰度图像特征到彩色图像权重的学习模型。通过训练图像像素间的相似关系,学习到最优权重,再利用学习到的权重将颜色从标记区域扩散到整张图片。实验结果表明,提出的方法能够获得更好的着色效果。