人懒,先放篇新闻,自动描线源码未放出,自动上色下载训练模型中…
新闻删掉了,有兴趣可以自行寻找
更新:
训练模型百度云地址:链接: 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
注意图片不要过大,否则速度会很慢
下面是我试用的几个例子:
由此可以看出,对于风景照效果还算可以,人像就有些差强人意了