前两天看到日本早稻田大学利用深度学习从原画中提取线稿的黑科技，发布了论文，但是并没有可以一试的源码。但是他的另一项黑科技给黑白照片上色，可以在git上下到试用的源码，不过，没有训练的源码。git地址：https://github.com/satoshiiizuka/siggraph2016_colorization
在开始试用之前，我们先要配置一下环境
以上都是需要安装的，本人是在OS X下安装的，大致步骤如下：
首先我们需要安装torch
# in a terminal, run the commands WITHOUT sudo
git clone https://github.com/torch/distro.git ~/torch --recursive
cd~/torch; bash install-deps;
./install.sh
先是从git上下载安装文件，下载完成后先安装LuaJIT，LuaRocks需要的环境，再安装LuaJIT，LuaRocks。安装完成后还会提示你将torch添加的你的path中，运行命令如下即可
# On Linux with bash
source~/.bashrc
# On Linux with zsh
source~/.zshrc
# On OSX or in Linux with none of the above.
source~/.profile
然后我们就可以用LuaRocks来安装torch,nn,image以及nngraph了
luarocks install nn
luarocks install image
luarocks install nngraph
环境配置完成之后我们先要下载官方训练完成的模型，可以通过下面的语句来下载
sh download_model.sh
也可以直接在http://hi.cs.waseda.ac.jp/~iizuka/data/colornet.t7下载
下载完成后就可以使用了，使用方法
th colorize.lua <input_image>[<output_image>]
比如
th colorize.lua ansel_colorado_1941.png out.png
注意一下，图片大小最好不要超过512x512px，不然会就较慢。
下面是我试玩的一些例子
从上面的例子可以看出，在风景图片的处理上表现还不错，但在人像上还稍显不足。
网友评论