Deep learning the GAN
Learn to generate the model through MNIST data, Anime data and other picture data.
The models mainly include: VAE, GAN, CGAN , DCGAN.
- numpy
- torch
- torchvision
- tqdm
- fire
Examples of Usage:
- Train:
mnist_dcgan.py
python mnist_dcgan.py train # If gpu is used
python mnist_dcgan.py train --use_gpu=False # If cpu is used
If your train is interrupted, you can continue training with "gen_epoch"
python mnist_dcgan.py train --gen_epoch=3
- Generate:
mnist_dcgan.py
python mnist_dcgan.py generate --use_gpu=False #If cpu is used
You can use "digit" to generate the number you want
python mnist_dcgan.py generate --gen_epoch=3 --digit=7