0
1

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 3 years have passed since last update.

stylegan2-ada のアニメ画像の学習済みモデルの公開(CPUで動作可能)

Last updated at Posted at 2021-08-10
1 / 2

はじめに

  • StyleGAN2 ada の検証とモデル公開
  • CPU でのデモとその手順

この記事の対象者

ディープラーニングモデルを手軽に実装したい方

デモ

モデルとコードはよ!
って方のためにこちらのGithubで公開しています。

必要な環境

  • python環境
  • git

手順

git clone https://github.com/syu-tan/yomaker.git
cd yomaker
pip install -r requirements.txt

学習済みのモデルを Google Drive からダウンロードして weights/ に配置する

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

00024-seed0001.png

out/seed0001.png に画像が生成されています。

学習は float16 で学習していますが、float32 に変換しています。
CPU で動作するので GPU の制約がなく、多くの人が手軽に使えると思います。
CUDA の環境構築とかほんと大変。。。。。

TODO

  • 512px モデルの公開
  • Dockerfile またはイメージの用意 

まとめ

やっぱりなんでも手軽さが大切だと思いました。
何番煎じですか?って感じですが自身で一から全てやってみたかったのでお許しください

Citation

NVIDIA StyleGAN2 ADA

License

Copyright © 2021, NVIDIA Corporation. All rights reserved.

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

0
1
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
0
1

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?