暇な日々にスパイスを

学んだ技術の備忘録

Deep Learningでモノクロ画像をカラーに!

Deep Learningが流行った時に読んだ記事の中で、モノクロをカラーにするというものに心惹かれていました。
それを実際に試してみました。

参考

どういったものかは、下記記事が参考になります。
qiita.com
道路のヒビがDeep Learningには芝生に見える、とか凄いワクワクものです。

導入

導入には下記記事を参考にしました。
qiita.com

途中色々と足りないライブラリを足しながら実行しました。

docker run -it ubuntu bash
git clone https://github.com/torch/distro.git ~/torch --recursive
cd ~/torch
sudo bash install-deps;
brew cmake
brew install cmake
./install.sh 
source ~/.bashrc
th
cd ..
git clone https://github.com/satoshiiizuka/siggraph2016_colorization.git
cd siggraph2016_colorization/
brew install wget
./download_model.sh 
luarocks install image
th colorize.lua input1.jpg output1.jpg

結果

画像はフリー モノクロで検索したモノを使用しました

ねこ
f:id:snona:20160801000136j:plain
f:id:snona:20160801000210j:plain

f:id:snona:20160801015900j:plain
f:id:snona:20160801015910j:plain

べんち
f:id:snona:20160801000230j:plain
f:id:snona:20160801000237j:plain

いい感じの雰囲気のやつ
f:id:snona:20160801000302j:plain
f:id:snona:20160801000312j:plain

そら
f:id:snona:20160801000342j:plain
f:id:snona:20160801000356j:plain

f:id:snona:20160801020028j:plain
f:id:snona:20160801020203j:plain

f:id:snona:20160801020208j:plain
f:id:snona:20160801020223j:plain

絵っぽいの
f:id:snona:20160801000412j:plain
f:id:snona:20160801000427j:plain

イラスト
f:id:snona:20160801000508j:plain
f:id:snona:20160801000540j:plain

絵っぽいの
f:id:snona:20160801000614j:plain
f:id:snona:20160801001127j:plain


f:id:snona:20160801015917j:plain
f:id:snona:20160801015924j:plain

夜景
f:id:snona:20160801015931j:plain
f:id:snona:20160801015940j:plain

くらげ
f:id:snona:20160801015947j:plain
f:id:snona:20160801015956j:plain

3年もののMacBook Airで実行しました。
処理中はPCがとても重いですが、このくらいの大きさなら1~5分程度で処理してくれます。

これ凄いですね。色々画像変換したいです。