けやみぃアーカイブ

CV勉強中の大学生のアウトプットです

2021-01-01から1年間の記事一覧

StyleGAN2で顔ランドマーク座標を指定して画像生成【後編】

GAN

前回はキーポイントを指定した際に他の顔属性も変わってしまっていたので,顔属性を保持しながらキーポイントを変化させるために属性推定器を学習させます. データ 顔属性 やはりラベル付きのデータが欲しいので,CelebAを使います.CelebAのラベルは40種類…

StyleGAN2で顔ランドマーク座標を指定して画像生成【前編】

GAN

PFNから発表された Surrogate Gradient Field for Latent Space Manipulation という論文の再現実装をしてみました.どんな論文かは以下の図を見てもらえるとわかるかと思います.上段は属性(年齢や性別など)を指定してStyleGAN2で生成した画像です.中段で…

PyTorchの異なるバージョン間での重み変換

なんやかんや複雑な事情で、学習時と推論時のPyTorchのバージョンが違った時の対処法です。特に学習時のバージョンの方が新しいと、推論する時に古いバージョンでは学習済みモデルを読み込んでくれないことがあるかと思います。まあ推論環境のPyTorchをアッ…