Progressive Growing of GANs inference in PyTorch with CelebA training snapshot
prog_gans_pytorch_inference
This is an inference sample written in PyTorch of the original Theano/Lasagne code.
I recreated the network as described in the paper of Karras et al. Since some layers seemed to be missing in PyTorch, these were implemented as well. The network and the layers can be found in model.py.
For the demo, a 100-celeb-hq-1024×1024-ours snapshot was used, which was made publicly available by the authors. Since I couldn’t find any model converter between Theano/Lasagne and PyTorch, I used a quick and dirty script to transfer the weights between the models (transfer_weights.py).
This repo does not provide the code for training the networks.
Simple inference
To run the demo, simply execute predict.py.
You can specify other weights with the --weights flag.
Example image:
