Ecosyste.ms: Awesome

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

Awesome Lists | Featured Topics | Projects

https://github.com/douglasrizzo/mygan


https://github.com/douglasrizzo/mygan

Last synced: 10 days ago
JSON representation

Awesome Lists containing this project

README

        

# Simple GAN implementations

Implementations of DCGAN and WGAN in PyTorch.

## Usage

Make sure you are already logged on to WandB, the scripts will create projects called "wgan_celeba" for you to track progress.

- Run `python -m mygan` to train a WGAN on CelebFaces.

## Samples

The table below displays samples generated from the same noise vectors at different epochs in training.

| 0 | 35 | 75 | 109 | 135 |
|-----------------|--------------------|--------------------|---------------------|---------------------|
| ![0](assets/image.png) | ![35](assets/image-1.png) | ![75](assets/image-2.png) | ![109](assets/image-3.png) | ![137](assets/image-4.png) |

You can see some cool samples [in this WandB report](https://wandb.ai/tetamusha/wgan_celeba/reports/GAN-training-on-image-datasets--Vmlldzo2NzMxODkx).

## References

-
-
-