Zenn
🦔

Windows11でStyleGAN3を試してみる

2022/07/27に公開
AI
Python
機械学習
PyTorch
tech

概要

Windows11でStyleGAN3の画像生成や学習を試してみる。

GitHub - NVlabs/stylegan3: Official PyTorch implementation of StyleGAN3

環境

  • OS : Windows11
  • GPU : RTX 3080Ti
  • Anaconda3

StyleGAN3を試す

準備

プロジェクトのダウンロード

git clone https://github.com/NVlabs/stylegan3.git
cd stylegan3

仮想環境作成

「environment.yml」を以下のように変更する

channels:
  - pytorch
  - nvidia
  - conda-forge <-追加

以下のコマンドで仮想環境構築&ライブラリインストール

conda env create -f environment.yml
conda activate stylegan3

conda install psutil

画像生成

画像生成

学習済みモデルを使用して画像生成

python gen_images.py --outdir=out --trunc=1 --seeds=0-3 ^
    --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-afhqv2-512x512.pkl




動画生成

python gen_video.py --output=lerp.mp4 --trunc=1 --seeds=0-31 --grid=4x2 ^
    --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-afhqv2-512x512.pkl

学習

データセットのダウンロード

MetFaces datasetをダウンロードする。

データセット作成

python dataset_tool.py --source=datasets/train_img --dest=datasets/train-256x256.zip --resolution=256x256

トレーニング開始

python train.py --outdir=training-runs --cfg=stylegan3-t --data=datasets/train-256x256.zip ^
    --gpus=1 --batch=16 --gamma=8.2 --mirror=1 --metrics=none

学習結果

  • kimg200
  • kimg400
  • kimg600

Discussion