Ecosyste.ms: Awesome

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

Awesome Lists | Featured Topics | Projects

https://github.com/thisisiron/tf2-gan

🐳 GAN implemented as Tensorflow 2.X
https://github.com/thisisiron/tf2-gan

gan gan-tutorial generative-adversarial-network tensorflow tensorflow-examples tensorflow-tutorials tensorflow2 tf2

Last synced: about 7 hours ago
JSON representation

🐳 GAN implemented as Tensorflow 2.X

Awesome Lists containing this project

README

        

# 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
- [X] GAN
- [X] CGAN
- [X] DCGAN
- [X] LSGAN
- [X] WGAN
- [X] WGAN-GP
- [X] CycleGAN
- [X] StarGAN
- [X] SRGAN
- [ ] SAGAN
- [ ] ACGAN
- [ ] infoGAN
- [ ] BEGAN
- [ ] BigGAN
- [ ] Stacked GAN
- [ ] EBGAN