We use a DCGAN on the CelebA dataset.
After cloning the repo on your local machine, first run the download.py
file to download and preprocess the data using the following command :
python download.py
You can easily change where the dataset and the results are saved in the constants.py
file.
To train the model, run the train.py
file using the following command:
python train.py
You can change the hyperparameters for the training and the model in the params.json
file.
Here are some samples obtained using the trained model :
You can download a pretrained model using the deconvolutional generator here
- Provide interactive notebook to show results