vinayakarannil / DCGAN

DCGAN implemetation for custom dataset

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

DCGAN

DCGAN implemetation for custom dataset. Inspired from Udacity deep learning nano degree programme.

Train for your own dataset

  1. Keep you images in "data" folder
  2. Run train.py script to start training the dcgan
 python train.py 

Model will resize all the images to 32*32. Checkpoints will be stored in "checkpoints" folder.

Test the model

  1. Run test.py
python test.py 

This will take load the checkpoint from checkpoints folder and generate random images.

alt text

References: Full paper here: http://arxiv.org/abs/1511.06434

About

DCGAN implemetation for custom dataset


Languages

Language:Python 100.0%