Play with pretrained GAN

Here is my jupyter notebook.
Let you to play with pretrained GAN (SAGAN + SNGAN + D2GAN)

GITHUB:

Requirement:

  • old fastai 0.7 (put in the same file)
  • python 3.6
  • pytorch 0.4
  • test on windows 10, cuda 9.0

change to your dir of dataset: put your image in file train

It is ok without any dataset and just play the pretrained GAN
image

Train on Guassian but generate on uniform will make the image looks better,
but the uniform images look more similar.

Continues change on row and colume

multi_noise display

show different results in mult std
small std make image looks more similar,
large std make image looks more difference.

4 Likes

Hello. I have spent several hours today trying to retrofit this code into the newest fastai package and finally losing my mind and giving up. Has anyone here tried to generate faces with a GAN and a pretrained network using fastai?
Thank you…