【StyleGAN】StyleGAN2-adaの画像生成をWindows11で試す

環境

  • Windows11
  • Anaconda3
  • Python3.7.11
  • NVIDIA GeForce GTX 1060 6GB

ライブラリのインストール

conda install tensorflow-gpu==1.14.0
conda install pillow
conda install requests

※tensorflowのver.を1.15にした際はうまくいかなかった。

stylegan2-adaをclone

git clone https://github.com/NVlabs/stylegan2-ada.git

人の顔の画像を生成

コマンド

python stylegan2-ada/generate.py --outdir=out --trunc=1 --seeds=85,265,297,849 --network=https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada/pretrained/metfaces.pkl

結果

f:id:sanpei07:20220131012307p:plain

猫の画像を生成

コマンド

python stylegan2-ada/generate.py --outdir=out --trunc=1 --seeds=85,265,297,849 --network=https://nvlabs-fi-cdn.nvidia.com/stylegan2-ada/pretrained/afhqcat.pkl

結果

f:id:sanpei07:20220131012413p:plain