Skip to content

Latest commit

 

History

History
143 lines (78 loc) · 3.27 KB

README.md

File metadata and controls

143 lines (78 loc) · 3.27 KB
Raw

yomaker

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

画像1

嫁画像をたくさん作るぞ〜！ってことで 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

画像2

  • スタイルミキシング
python style_mixing.py --outdir=out --rows=85,100,75,458,1500 --cols=55,821,1789,293 --network=weights/00024-256px-animeportrait.pkl

画像4

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

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

proj.mp4

詳細

  • モデル

    NVIDIA StyleGAN2 ADAを使用

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

  • ログ結果の表示 画像3

    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 とスタイルミックス

    画像1 画像2

    画像4

加えて、計算資源が余っている時に 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.