/gans-collection

Pytorch implementations of Generative Adversarial Networks variants like GAN, CGAN, DCGAN, and Pix2Pix. This repository also includes brief summaries of some machine learning papers that have been used to develop models.

Primary LanguageJupyter Notebook

Gans Collection

Pytorch implementations of Generative Adversarial Networks variants like GAN, CGAN, DCGAN, and Pix2Pix. This repository also includes brief summaries of some machine learning papers that have been used to develop models.

Installation

$ git clone https://github.com/hungle45/gans-collection.git
$ pip install -r requirements.txt

Experiment

Sketch2Color Translation

Anime Sketch Colorization Pair Dataset

For this task, Anime Sketch Colorization Pair dataset is used. You can download it from Kaggle link.

Pix2Pix

Trainnig logs


Epoch 1


Epoch 10


Epoch 20


Epoch 30

Result





1st col: Input / 2nd col: Generated / 3rd col: Target

Digits Generation

For this task, MNIST dataset is used. You can download it from Kaggle link.

GAN

Result


Generated digits by GAN

DCGAN

Result


Generated digits by DCGAN

CGAN

Result


Generated digits by CGAN with conditions 0..9 (from left to right)