Home

Awesome

chainer-gogh

Implementation of "A neural algorithm of Artistic style" (http://arxiv.org/abs/1508.06576)

解説記事: https://research.preferred.jp/2015/09/chainer-gogh/

<img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/cat.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/style_0.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/im0.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/style_1.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/im1.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/style_2.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/im2.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/style_3.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/im3.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/style_4.jpg" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/im4.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/style_5.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/im5.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/style_6.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/im6.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/style_7.png" height="150px"> <img src="https://raw.githubusercontent.com/mattya/chainer-gogh/master/sample_images/im7.png" height="150px">

(VGG, lam=0.0075, after 5000 iteration)

Usage:

Chainerをインストール

pip install chainer

詳しくはhttps://github.com/pfnet/chainer

モデルをダウンロード

お手軽。(-m nin)

きれいな絵がかけるがとても重い。(-m vgg, -m vgg_chainer) vgg_chainerではモデルのダウンロードの必要はなく、初回を除いて非常に高速でロードできるようになります(chainer 1.19以降で動作)。

NIN並に軽く、ポテンシャルもあるはずだが、最適なパラメタがわかっていない。(-m googlenet)

VGGより軽く、二次元画像にとても強いはずだが、最適なパラメタがわかってない。(-m i2v)

CPU実行

python chainer-gogh.py -m nin -i input.png -s style.png -o output_dir -g -1

GPU実行

python chainer-gogh.py -m nin -i input.png -s style.png -o output_dir -g GPU番号

VGG実行サンプル

python chainer-gogh.py -m vgg_chainer -i input.png -s style.png -o output_dir -g 0 --width 256

モデルの指定方法

-m nin

のninを、vgg, vgg_chainer, googlenet, i2vに切り替えることが可能。 モデルファイルはディレクトリ直下に置いて、デフォルトの名前のまま変えないこと。

複数枚同時生成

input0.png style0.png
input1.png style1.png
...

そして、chainer-gogh-multi.pyの方を実行

python chainer-gogh-multi.py -i input.txt

VGGを使うときはGPUのメモリ不足に注意

パラメタについて

注意