Skip to content

thisisiron/TF2-GAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

74 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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