什么?AI黑白图像能够自动上色?( 二 )

作者 | 李秋键 责编 | 刘静

本篇文章将利用GAN算法的pixpix模型实现对黑白图片进行上色 。 下面简单介绍一下图片像素知识 , 和黑白图像一样 , 彩色图像中的每一层 , 也有~的值 。 值表示这个层中没有颜色 。 如果像素网格所有颜色通道的值都为 , 那么这个图像像素就是黑色的 。 神经网络在输入值和输出值之间创建了一种关系 。 为了能够更为准确地完成着色任务 , 网络需要找到能够将灰度图像和彩色图像联系起来的特征 。

总的来说就是 , 我们需要找到能够将灰度值网格链接到三个颜色网格的特征 。

f()是神经网络 , [BW
是我们的输入 , [R
、[G
、[B
是我们的输出

现在 , 随着数据集的增加 , 由于我们处理的是高分辨率图像 , 因此我们需要更多的计算能力 。 为此 , 我们使用 Deep Cognition 的 Deep Learning Studio jupyter notebooks , 它为Amazon 的深度学习示例提供了GPU , 可用来训练模型 。

推荐阅读