本日は画像をジブリアニメ風に加工することのできる
CartoonGANというものを試してみました。

解説動画はこちら



さて
まずはGANは

敵対的生成ネットワーク
英: Generative adversarial networks

と呼ばれるもので
画像を用いた学習を行うことによって
様々なスタイルの画像を
生成できるというものです。

沢山の種類がありますが
今回はジブリアニメを学習に用いた
Cartoon-GANというものを用います。
cartoon-gan

ここのリンクからソースなどを
ダウンロードする事ができます。

動かすには他に
Anacondaや各種ライブラリが必要です。
opencv,tqdm,pillow,numpy
python -m pip install tqdm pillow numpy matplotlib opencv-python

Pytorch
conda install pytorch torchvision -c pytorch

ZIPを展開すると
中にpredict.pyというファイルがあるので
GPUが使えないマシンを使用する方は
以下の部分のコードをコメントアウトして
1行追加してもらう必要があります。

# Load weights
#netG.load_state_dict(torch.load(pretrained_dir))
netG.load_state_dict(torch.load(pretrained_dir,map_location=torch.device('cpu')))

さてこれでコードの準備はできました。

実行方法は
predict.pyのあるディレクトリに移動し

python predict.py -i 画像の場所 -o 変換後の画像の場所 -d cpu

となります。

今回の検証では10枚ほどを
変換して試してみました。

download

download-2
download-1

なかなか面白いですね
全体的に青み、緑みが強めになり
服などはアニメ風に塗りが行われ
ジブリっぽいところもあります。

他の画像の変換については
是非動画の方でご覧くださいませ。

他のアニメ風にスタイル変換する
GANもあるようですので
他のも試してみたいですね。

今回はこれまでです
それでは。