chainer-gogh

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

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

(VGG, lam=0.0075, after 5000 iteration)

Usage:

Chainerをインストール

pip install chainer

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

モデルをダウンロード

• NIN (お手軽) https://gist.github.com/mavenlin/d802a5849de39225bcc6

• VGG (きれいな絵がかけるがとても重い) https://gist.github.com/ksimonyan/211839e770f7b538e2d8#file-readme-md

CPU実行

python chainer-gogh.py -i input.png -s style.png -o output.png -g -1

GPU実行

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

VGG実行サンプル

python chainer-gogh.py -i input.png -s style.png -o output.png -g 0 -m VGG_ILSVRC_16_layers.caffemodel --width 256

複数枚同時生成

• まず、input.txtというファイル名で、以下の様なファイルを作る。

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

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

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

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

パラメタについて

• --lr: 学習速度。生成の進捗が遅い時は大きめにする
• --lam: これを上げるとinput画像に近くなり、下げるとstyle画像に近くなる

注意

• 現在のところ画像は正方形に近いほうがいいです