diff --git a/README.md b/README.md index 28aa13a..1ec951f 100644 --- a/README.md +++ b/README.md @@ -2,15 +2,21 @@ 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]( +https://github.com/ivclab/BigGAN-Generator-Pretrained-Pytorch/releases/latest) # Demo +- To run the code, please download the pretrained weights first. ```shell python demo.py -w [-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 |![alt text](./assets/p1.png)| @@ -22,7 +28,14 @@ python demo.py --pretrained_weight ./biggan128-release.pt --size 128 --truncatio |*class 821 (512 x 512)*| # Dependencies -Please refer to the environment.yml file. +dependencies: + - python=3.6 + - cudatoolkit=10.0 + - pytorch + - torchvision + - scipy + +Please also refer to the environment.yml file. # Pretrained Weights The pretrained weights are converted from the tensorflow hub modules: