HitLuca / GAN-collection

my own take on implementing various GANs

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GAN-collection

License MIT HitCount Total alerts Language grade: Python

Description

Collection of various GAN models using my own way of implementing them.

Models

At this point four different generative models are implemented:

  • Conditional Improved Wasserstein GAN (CWGAN-GP)
  • Improved Wasserstein GAN (WGAN-GP)
  • Boundary Equilibrium GAN (BEGAN)
  • Deep Convolutional GAN (DCGAN)
  • Variational AutoEncoder with learned similarity metric (WGAN-GP-VAE)

The WGAN-GP-VAE model implementation comes from my other project, as a part of my Master Thesis (AI)

Project structure

The models folder contains all the generative models implemented.

Prerequisites

To install the python environment for this project, refer to the Pipenv setup guide

About

my own take on implementing various GANs

License:MIT License


Languages

Language:Python 100.0%