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: 3 days ago
JSON representation

🐳 GAN implemented as Tensorflow 2.X

• Host: GitHub
• URL: https://github.com/thisisiron/tf2-gan
• Owner: thisisiron
• License: mit
• Created: 2019-07-16T10:12:53.000Z (over 5 years ago)
• Default Branch: master
• Last Pushed: 2020-11-18T04:08:44.000Z (almost 4 years ago)
• Last Synced: 2024-11-12T17:03:43.903Z (3 days ago)
• Topics: gan, gan-tutorial, generative-adversarial-network, tensorflow, tensorflow-examples, tensorflow-tutorials, tensorflow2, tf2
• Language: Python
• Homepage:
• Size: 3.69 MB
• Stars: 64
• Watchers: 5
• Forks: 19
• Open Issues: 4
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

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