Skip to content

Latest commit

 

History

History
23 lines (18 loc) · 679 Bytes

README.md

File metadata and controls

23 lines (18 loc) · 679 Bytes

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