NVIDIA DIGITS 6のPretrained ModelでGANを試してみた(後編)

digits6-pretrained-gan-demo

NVIDIA DIGITS 6のModel Storeからダウンロードしたceleb-a-gan, celeb-a-gan-encoderを使用して、GAN (Generative Adversarial Network)を試してみました。

CelebA Dataset、CelebA GAN Modelの作成については、以下の記事(前編)をご覧ください。

NVIDIA DIGITS 6のPretrained ModelでGANを試してみた(前編)
NVIDIA DIGITS 6のModel Storeからダウンロードしたceleb-a-gan, celeb-a-gan-encoderを使用して、GAN (Generative Adversarial Network)を試してみました。
スポンサーリンク

DIGITS 6

CelebA GAN Encoder

前編で作成したCelebA GAN Encoder(ここでは「CelebA64-GAN-Encoder」)ページに移動します。

CelebA GAN Encoder
上の図のように、「Select Visualization Method」は「GAN」、「Visualization Options」の上側は「CelebA Encoder」を選択し、本家のREADME.mdにはないのですが、「Attributes vector file」に前編で作成したattribute vectors(ここでは、/data/digits/CelebA64-attribute/attributes_z.pkl)を設定します。

更に、「Select Inference form」は「GAN」、「GAN inference Options」は上から順に「CelebA」と「CelebA – Encode list of images」を選択し、「File list」と「Image folder」にそれぞれlist_attr_celeba.txtへのパス(ここでは/data/digits/CelebA/list_attr_celeba.txt)とCelebA Datasetを展開した場所(ここでは/data/digits/CelebA/img_align_celeba)を設定します。

また、「Number of images to encode」にエンコードしたいイメージ数を設定します。

「Test」ボタンをクリックしてしばらく待つと、以下のような出力が得られます。
なお、右側の数列がZ vectorとなります。

CelebA GAN Encoder Inference

CelebA GAN

次に、同じく前編で作成したCelebA GAN(ここでは「CelebA64-GAN」)ページに移動します。

CelebA GAN Image
上の図のように、「Select Visualization Method」は「Image output」、「Visualization Options」は上から順に「RGB」「HWC」「Normalize」「No」を選択します。

「Select Inference form」は「GAN」、「GAN inference Options」は上から順に「CelebA」と「CelebA – add/remove attributes」を選択し、「Attributes vector file」に前編で作成したattribute vectorsを設定します。

「Z vector」には、CelebA GAN Encoderの右側の数列([]の内側の数列のみ)を設定できます。

「Attributes Params」を設定して、「Test」ボタンをクリックすると、以下のような出力が得られます。

CelebA GAN Image Inference
また、CelebA GAN(ここでは「CelebA64-GAN」)ページで、以下の図のように、「Select Visualization Method」は「GAN」、「Visualization Options」は「Grid」を選択します。

更に、「Select Inference form」は「GAN」、「GAN inference Options」は上から順に「CelebA」と「CelebA – Analogy」を選択し、出力の左上、右上、左下に対応する「Z vector」を設定して、「Test」ボタンをクリックします。

CelebA GAN Grid
すると、以下のような出力が得られます。

CelebA GAN Grid Inference
上の図のAnimationの部分は、実際には下の図のようなアニメーションGIFが表示されています。

CelebA GAN Grid Inference Animation

まとめ

NVIDIA DIGITS 6のModel Storeからダウンロードしたceleb-a-gan, celeb-a-gan-encoderを使用して、GAN (Generative Adversarial Network)を試してみました。