yomaker

• 学習途中ですが、２次元イラストを学習させたのでモデルとデモの公開

嫁画像をたくさん作るぞ〜！ってことで yomake

GAN とかはマシンスペック求められたりするので気軽に楽しめない人のために

とりあえず、動かしたい方は 以下へ

環境構築

python version => 3.8

pip install -r requirements.txt

学習済みモデル(.pkl)をダウンロードして weghts/ に格納する。

OSX(Mac) と Ubuntu にて CPU 動作確認

デモ

seed を変更すると出力画像が変化します。

out/ に画像が生成されます。

• 単一画像生成

python generate.py --outdir=out --trunc=1 --seeds=56,1 --network=./weights/00024-256px-animeportrait.pkl

• スタイルミキシング

python style_mixing.py --outdir=out --rows=85,100,75,458,1500 --cols=55,821,1789,293 --network=weights/00024-256px-animeportrait.pkl

• 類似画像探索

python projector.py --outdir=out --target={探索したいい画像のパス} --network=weights/00024
-256px-animeportrait.pkl

VGG16 でレイテントを近似します。(なので時間かかります)

proj.mp4

詳細

• モデル

  NVIDIA StyleGAN2 ADAを使用

  事前学習モデルは half(floast16) で学習して CPU 実行のために float32 に重みを変換しています。

• ログ結果の表示

  tensorboard を使用して手元で動かす場合

  pip install tensorboard
  tensorboard --logdir=results/

  http://localhost:6006を参照

  論文通り FID 10 あたりまで達成。 しかし、細部にはアーティファクトが目立つ

• データ

  twitter のイラスト画像(danboru)

  画像枚数: 1711752 枚

  NVIDIA RTX 3090 で 6日ほど

アップデート

現在学習中ですので、256px は FID は良くなります。

1. 2021/08/10 にてモデルを更新しました。

• 過去画像一覧

  初期の seed 56,1 とスタイルミックス

  

  

加えて、計算資源が余っている時に 512px もうまくいけば公開します。

感謝

NVIDIA にはマシンといい、モデルフレームといい大変お世話になりました。

　　*☆**☆*
　*☆*★★★*☆*
`*☆*★※※※★*☆*
*☆★※☆☆☆※★☆*
*☆ｷﾀ━(ﾟ∀ﾟ)━!!☆*　　　有難うございます！
*☆★※☆☆☆※★☆*
　*☆*★※※※★*☆*
　*☆*★★★*☆*
　　*☆**☆*

Citation

NVIDIA StyleGAN2 ADA

License

Copyright © 2021, NVIDIA Corporation. All rights reserved.

This work is made available under the Nvidia Source Code License.