python 自动上色_Github | 线稿自动上色

今天要介绍的是一个 Github 项目，项目地址如下，它实现了对线稿的自动上色功能，效果还是很不错的，一起来看看吧。

https://github.com/SerialLain3170/Colorization

简介

本项目主要实现了自动将线稿转成彩色图片的功能。当然，我们可以只训练处理线稿的神经网络，但实际应用中我们需要提前用指定颜色给线稿上色的功能。实现上色的方法有很多，包括给定提示(hint)。

不带提示

没有提示的上色方法

输入：仅线稿

Atari

带有提示的上色方法，提示一般是在特定区域所需颜色的线条(比如 PaintsChainer)

输入：线稿和 atari

标签

提示是标签的上色方法

输入：线稿和标签

参照物

采用参照图片作为提示的上色方法(比如 style2paints V1)

输入：线稿和参考图片

线提取方法

在线提取方法有很多改进版方法，比如 **XDoG **或者 SketchKeras。但是如果仅仅在一种类型的线稿上训练模型，模型会对这种类型的线稿过拟合，从而无法对其他类型的线稿实现自动上色功能。因此，和 Tag2Pix 一样，这里使用多种不同的线稿作为训练数据进行网络的训练。

使用的是下面三种类型的线：

XDoG：使用两个高斯分布的差异到标准差进行线提取；

SketchKeras：采用 UNet 进行线提取。通过这种方法提取得到的线会类似铅笔素描；

Sketch简化版：通过全卷积网络