zhangzhe
人懒,先放篇新闻,自动描线源码未放出,自动上色下载训练模型中…
新闻删掉了,有兴趣可以自行寻找


更新:
训练模型百度云地址:链接: http://pan.baidu.com/s/1jI8iKQi 密码: ses6
请务必检查MD5 (colornet.t7) = c88fa2bb6dc9f942a492a7dc7009b966

搞好了,试了几张图片,系统为osx 10.11.3,下面是过程:

首先按照文档要求配置环境:

安装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

这一步完成之后会提示你将torch添加到环境变量中,执行命令:

# On Linux with bash
source ~/.bashrc
# On Linux with zsh
source ~/.zshrc
# On OSX or in Linux with none of the above.
source ~/.profile

安装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 zhangzhe.jpg haha.jpg

注意图片不要过大,否则速度会很慢

下面是我试用的几个例子:

fengjing
赫本
行尸走肉
城市
街道
熊猫
菊花
804副本.jpg
803副本.jpg
802副本.jpg
801副本.jpg
pdh2.jpg
lhg副本.jpg
由此可以看出,对于风景照效果还算可以,人像就有些差强人意了

一些参考文档:
自动上色github地址
torch安装文档
自动描线技术论文(英文)