Skip to content
vinayakarannilPublic

About

DCGAN implemetation for custom dataset

Topics

Resources

Stars

6 stars

Watchers

0 watching

Forks

Latest commit

 

History

2 Commits

Folders and files

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

About

DCGAN implemetation for custom dataset

Topics

Resources

Stars

6 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages