StyleGAN2 TensorFlowの重みをPytorchに変換する


PytorchのStyleGANで顔画像を生成する方法です

そろそろStyleGANをいじってみたい

StyleGAN2の公式リポジトリはTensorFlow1。
Pytorchに変換して使ってみます。

stylegan2-pytorch

公式リポジトリとpytorchバージョンのリポジトリをクローンします。

オリジナルのチェックポイントファイルpklをpytorch形式に変換します。
事前トレーニング済みの重みは、アニメなどいろいろあります。

python convert_weight.py --repo ../stylegan2 {path_to_checkpoint.pkl}

Pytorchの重みptファイルと、オリジナルとpytorchバージョンの画像生成比較pngファイルが保存されます。

オリジナルと変換バージョンで同じ出力がされています。

変換された重みptで画像を生成します。

python generate.py --sample 1 --pics 1 --ckpt {converted_checkpoint_path.pt}

🐣


フリーランスエンジニアです。
お仕事のご相談こちらまで
[email protected]

Core MLやARKitを使ったアプリを作っています。
機械学習/AR関連の情報を発信しています。

Twitter
Medium