This repository demonstrates how to generate images of handwritten digits using a Deep Convolutional Generative Adversarial Network (DCGAN).
An implementation of 2015 DCGAN Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks
Images generated during model training
file name | content |
---|---|
dataset_utils.py | |
train_dcgan_model.py | |
inference_by_dcgan_model.py | |
training_checkpoints | Save model weight |
train_epoch_produce_images | The picture generated during the training of the model is generated one per training period. Used to check the effect of the model to generate a picture. |
inference_produce_images | Save images generated using trained models |
- python 3+
- tensorflow version 2
Training_checkpoints already has trained model weights that can be used directly to execute 3. Model Inference.
python dataset_utils.py
python train_dcgan_model.py
python inference_by_dcgan_model.py
EPOCHS = 50