Skip to content

vinayakarannil/DCGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Releases

No releases published

Packages

No packages published

Languages