はじめに
一枚の自撮り画像を用いて、性別や年齢、笑顔、髪の毛、口や鼻、メガネなどのパラメータを変化させて、顔画像を生成するALAEをやってみました。
システム環境
- Windows10（RTX2080 Max-Q、i7-8750H、RAM16GB）
- Anaconda 2020.02
- Python 3.7
導入
ALAEをクローンします。
ALAE用の環境を作成します。
$ cd ALAE-master
$ conda create -n ALAE python=3.7
$ conda activate ALAE
ライブラリをインストールします。
$ pip install torch==1.5.0 torchvision==0.6.0 -f https://download.pytorch.org/whl/torch_stable.html
$ pip install -r requirements.txt
下記を実行し、学習済みモデルをダウンロードします。
$ python training_artifacts/download_all.py
デモを実行します。
$ python interactive_demo.py
ALAE-master\dataset_samples\faces\realign1024x1024に自撮り画像を置いて実行してみました。
| 入力 | 出力 |
|---|---|
 |
 |
かわいいｗ
ALAE : Adversarial Latent Autoencoders Demo #CVPR2020 #AI #画像処理 #DeepLearning #機械学習 #CV pic.twitter.com/aUf4pWV5KZ
— 藤本賢志(ガチ本)@pixivFANBOXはじめました (@sotongshi) April 27, 2020
お疲れ様でした。