GANs-collections-tf2.0_keras-eager_mode
GANs-collections-tf2.0_keras-eager_mode copied to clipboard
This repository implements all kinds of GAN-models based on tensorflow2.0 keras API including GAN, CGAN, WGAN, WGAN_GP, VAE, CVAE, LSGAN, infoGAN, EBGAN, BEGAN, ACGAN
Description
This repository implements all kinds of GAN-models based on tensorflow2.0 keras API and in eager mode. The code was only validated on windows platform and just cpu mode.
How to use
Install tensorflow2.0-aplha version. Anaconda virtual enviroment is recommended. To support GPU mode,please refer to this(I haven't validated on GPU beacuse have no GPU to use).
TODO
- [x] To implement ACGAN, EBGAN, BEGAN, etc.
- [x] Support GPU run.