Skip to content

Latest commit

 

History

History
44 lines (39 loc) · 1000 Bytes

README.md

File metadata and controls

44 lines (39 loc) · 1000 Bytes
Raw

TF2-GAN

Introduction

Typical GANs are implemented as Tensorflow 2.
I followed the suggestions in the papers, and I slightly changed the model structure or optimizer for simple task.

Requirements

Tensorflow 2.0
Tensorflow Datasets
Tensorflow-addons

How to Run

cd GAN_DIR_YOU_WANT
python train.py

File structure

Name Description
utils.py Loss function, Image storage function, etc.
model.py Model Architecture
train.py Model learning and Loading datasets

GAN List

  • GAN
  • CGAN
  • DCGAN
  • LSGAN
  • WGAN
  • WGAN-GP
  • CycleGAN
  • StarGAN
  • SRGAN
  • SAGAN
  • ACGAN
  • infoGAN
  • BEGAN
  • BigGAN
  • Stacked GAN
  • EBGAN