GANs

KO | EN

gitlite — search

#javascript #python #hacktoberfest #react #ai #typescript #llm #go #golang #android #machine-learning #rust #deep-learning #linux

★ 8 Open GitHub ↗

Implementations of WGAN and WGAN-GP trained on CIFAR-10 and Tiny Imagenet

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# GANs
### Swiss Roll Dataset
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Swiss%20Roll/swiss%20roll.png)

### Vanilla WGAN
#### Optimizer: RMSProp, Learning Rate: 0.00005, n_critic = 5
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Swiss%20Roll/WGAN.gif)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Swiss%20Roll/WGAN.png)

### WGAN with Gradient-Penalty
#### Optimizer: Adam, Learning Rate: 0.00005, n_critic = 5
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Swiss%20Roll/WGAN-GP.gif)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Swiss%20Roll/WGAN-GP.png)

### CIFAR-10
#### Algorithm: WGAN-GP, Optimizer: Adam, Learning Rate: 0.0001, n_critic = 5
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/CIFAR-10/cifar-wgan-gp.gif)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/CIFAR-10/wgan-gp-cifar.png)

#### Samples:
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/CIFAR-10/cifar-samples/1.png)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/CIFAR-10/cifar-samples/3.png)

#### Trained for 200k iterations
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/CIFAR-10/Fully%20trained/Training.png)

#### Samples:
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/CIFAR-10/Fully%20trained/samples/17.png)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/CIFAR-10/Fully%20trained/samples/2.png)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/CIFAR-10/Fully%20trained/samples/6.png)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/CIFAR-10/Fully%20trained/samples/9.png)

### Tiny ImageNet
#### Algorithm: WGAN-GP, Optimizer: Adam, Learning Rate: 0.0001, n_critic = 5
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Tiny%20ImageNet/disc_losses.png)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Tiny%20ImageNet/gen_losses.png)

#### Samples:
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Tiny%20ImageNet/tiny-imagenet-samples/11.png)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Tiny%20ImageNet/tiny-imagenet-samples/14.png)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Tiny%20ImageNet/tiny-imagenet-samples/18.png)
![alt text](https://github.com/sudeepkatakol/GANs/blob/master/Tiny%20ImageNet/tiny-imagenet-samples/9.png)

GANs

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README