BigGAN Generators with Pretrained Weights in Pytorch

Pytorch implementation of the generator of Large Scale GAN Training for High Fidelity Natural Image Synthesis (BigGAN).

Download Pretrained Weights

The Pretrained weights can be downloaded from the latest release. link

Dependencies

dependencies:

python=3.6
cudatoolkit=10.0
pytorch
torchvision
scipy

Please also refer to the environment.yml file.

Demo

To run the code, please download the pretrained weights first.

python demo.py -w <PRETRAINED_WEIGHT_PATH> [-s IMAGE_SIZE] [-c CLASS_LABEL] [-t TRUNCATION] 
python demo.py -w ./biggan512-release.pt -s 512 -t 0.3 -c 156 
python demo.py -w ./biggan256-release.pt -s 256 -t 0.02 -c 11 
python demo.py --pretrained_weight ./biggan128-release.pt --size 128 --truncation 0.2 --class_label 821

Valid image size: 128, 256, 512
Valid class label: 0~999
Valid truncation: 0.02~1.0

Results


class 156 (512 x 512)

class 11 (512 x 512)

class 821 (512 x 512)

Pretrained Weights

The pretrained weights are converted from the tensorflow hub modules:

References

paper: https://arxiv.org/abs/1809.11096

https://github.com/ajbrock/BigGAN-PyTorch

Contact

Please feel free to leave suggestions or comments to Tsung-Hung Hsieh(andrewhsiehth@gmail.com).