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関連の情報を発信しています。
Author And Source
この問題について(StyleGAN2 TensorFlowの重みをPytorchに変換する), 我々は、より多くの情報をここで見つけました https://qiita.com/john-rocky/items/b77938f2af8c31241888著者帰属:元の著者の情報は、元のURLに含まれています。著作権は原作者に属する。
Content is automatically searched and collected through network algorithms . If there is a violation . Please contact us . We will adjust (correct author information ,or delete content ) as soon as possible .