はじめに

この記事の対象者

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

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

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

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

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

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

Copyright © 2021, NVIDIA Corporation. All rights reserved.

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