python之深度学习自动上色

同学在空间分享的画

昨天下午看到同学在QQ空间发这一张黑白画，并留言道“”上色很痛苦的/ll“，于是我突然想到前几天在微信某公众号看到的深度学习上色技术，于是百度了一下，果然在github上有源码，实现这个项目很简单，在终端中执行如下命令

$ git clone -b master --single-branch https://github.com/richzhang/colorization.git
$ ./models/fetch_release_models.sh

然后，在终端运行python ./colorize.py -img_in [[INPUT_IMG_PATH]] -img_out [[OUTPUT_IMG_PATH]]即可，例如：

python ./colorize.py -img_in ./demo/imgs/ILSVRC2012_val_00041580.JPEG -img_out ./out.png

test:

上色前

上色后

测试效果不错，于是我开始处理同学发的那张图片：

1、将图片进行剪切，留下图画的部分;
2、考虑到这位同学在空间发的是手机拍的图片，于是我又将其转换为灰度图;

下面可以进行上色了：

上色前

上色后

效果不散很理想，也许是这张图黑白过于明显，几乎二值图，所以无法上色。如果是素描作品的画，也许效果会好一点，哈哈。