Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/rahulbhalley/progressive-growing-of-gans.pytorch

ProGAN with Standard, WGAN, WGAN-GP, LSGAN, BEGAN, DRAGAN, Conditional GAN, InfoGAN, and Auxiliary Classifier GAN training methods
https://github.com/rahulbhalley/progressive-growing-of-gans.pytorch

acgan auxiliary-classifier-gan began conditional-gan discriminator dragan generative-adversarial-network generator infogan infogan-pytorch lsgan lsgan-pytorch optimal-transport progressive-gan pytorch tensorboard wasserstein-gan wgan wgan-gp wgans

Last synced: about 1 month ago
JSON representation

ProGAN with Standard, WGAN, WGAN-GP, LSGAN, BEGAN, DRAGAN, Conditional GAN, InfoGAN, and Auxiliary Classifier GAN training methods

Host: GitHub
URL: https://github.com/rahulbhalley/progressive-growing-of-gans.pytorch
Owner: RahulBhalley
License: mit
Created: 2017-10-30T02:07:18.000Z (over 7 years ago)
Default Branch: master
Last Pushed: 2024-09-06T13:56:29.000Z (5 months ago)
Last Synced: 2024-12-10T03:32:11.041Z (about 2 months ago)
Topics: acgan, auxiliary-classifier-gan, began, conditional-gan, discriminator, dragan, generative-adversarial-network, generator, infogan, infogan-pytorch, lsgan, lsgan-pytorch, optimal-transport, progressive-gan, pytorch, tensorboard, wasserstein-gan, wgan, wgan-gp, wgans
Language: Python
Homepage: https://arxiv.org/abs/1710.10196
Size: 17.5 MB
Stars: 51
Watchers: 5
Forks: 5
Open Issues: 5
Metadata Files:
- Readme: README.md
- License: LICENSE

Awesome Lists containing this project

README

# Progressive Growing of Generative Adversarial Network

This is PyTorch implementation of Progressive Growing GANs. The network is trainable on custom image dataset.

Place your dataset folder inside `data` folder. The training stats are added to `repo` folder as the training progresses.

### Training Configuration

The network training parameters can be configured with following flags.

#### General settings

- `--train_data_root` - Set your data directory
- `--random_seed` - Random seed to reproduce the experiments
- `--n_gpu` - Number of GPUs for multiple GPU training

#### Training parameters

- `--lr` - Learning rate
- `--lr_decay` - Learning rate decay at every resolution transition
- `--eps_drift` - Coefficient for the drift loss
- `--smoothing` - Smoothing factor for smoothed generator
- `--nc` - Number of input channels
- `--nz` - Input dimension of noise
- `--ngf` - Feature dimension of final layer of generator
- `--ndf` - Feature dimension of first layer of discriminator
- `--TICK` - 1 tick = 1000 images = (1000/batch_size) iteration
- `--max_resl` - Maximum resolution (10-->1024, 9-->512, 8-->256)
- `--trns_tick` - Transition tick
- `--stab_tick` - Stabilization tick
- `--gan_type` - GAN training methodology (choices: 'standard', 'wgan', 'wgan-gp', 'lsgan', 'began', 'dragan', 'cgan', 'infogan', 'acgan')
- `--lambda_gp` - Gradient penalty lambda for WGAN-GP
- `--lambda_drift` - Drift loss coefficient for WGAN
- `--gamma` - Equilibrium constant for BEGAN
- `--lambda_k` - Learning rate for k in BEGAN
- `--lambda_info` - Information loss weight for InfoGAN
- `--n_classes` - Number of classes for conditional GANs

#### Network structure

- `--flag_wn` - Use of equalized-learning rate
- `--flag_bn` - Use of batch-normalization (not recommended)
- `--flag_pixelwise` - Use of pixelwise normalization for generator
- `--flag_gdrop` - Use of generalized dropout layer for discriminator
- `--flag_leaky` - Use of leaky ReLU instead of ReLU
- `--flag_tanh` - Use of tanh at the end of the generator
- `--flag_sigmoid` - Use of sigmoid at the end of the discriminator
- `--flag_add_noise` - Add noise to the real image(x)
- `--flag_norm_latent` - Pixelwise normalization of latent vector (z)
- `--flag_add_drift` - Add drift loss

#### Optimizer setting

- `--optimizer` - Optimizer type
- `--beta1` - Beta1 for Adam optimizer
- `--beta2` - Beta2 for Adam optimizer

#### Display and save setting

- `--use_tb` - Enable tensorboard visualization
- `--save_img_every` - Save images every specified iteration
- `--display_tb_every` - Display progress every specified iteration

### GPU Note

Make sure your machine has CUDA enabled GPU(s) if you want to train on GPUs. Change the `--n_gpu` flag to positive integral value <= available number of GPUs.

### GAN Training Methodologies

This implementation supports multiple GAN training methodologies:

1. Standard GAN (default)
2. Wasserstein GAN (WGAN)
3. Wasserstein GAN with Gradient Penalty (WGAN-GP)
4. Least Squares GAN (LSGAN)
5. Boundary Equilibrium GAN (BEGAN)
6. DRAGAN
7. Conditional GAN (CGAN)
8. InfoGAN
9. Auxiliary Classifier GAN (ACGAN)

To select a specific training methodology, use the `--gan_type` flag:

```
python main.py --gan_type standard
python main.py --gan_type wgan
python main.py --gan_type wgan-gp
python main.py --gan_type lsgan
python main.py --gan_type began
python main.py --gan_type dragan
python main.py --gan_type cgan
python main.py --gan_type infogan
python main.py --gan_type acgan
```

### Related Links

- [Progressive Growing of GANs for Improved Quality, Stability, and Variation](https://arxiv.org/abs/1710.10196)
- [Original Implementation in Lasagne/Theano](https://github.com/tkarras/progressive_growing_of_gans)
- [Wasserstein GAN](https://arxiv.org/abs/1701.07875)
- [Improved Training of Wasserstein GANs](https://arxiv.org/abs/1704.00028)