Pete43 / GANs-collections-tf2.0_keras-eager_mode

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

• To implement ACGAN, EBGAN, BEGAN, etc.
• Support GPU run.