dcgan

A PyTorch implementation of Unsupervised Representation Learning With Deep Convolutional Generative Adversarial Networks

Installation

git clone https://github.com/andregaio/dcgan.git
cd dcgan
conda create -n dcgan python=3.8
conda activate dcgan
pip install -r requirements.txt

Dataset

Celeb-A Faces has been used for training. Download and unzip 'img_align_celeba.zip' from Google Drive into the 'dataset' folder

Models

• DCGAN

Training

python dcgan/train.py

Inference

python dcgan/infer.py --weights weights/last_weights.pt --out test/image.png

Results

Notes

• This is a refactored implementation of https://pytorch.org/tutorials/beginner/dcgan_faces_tutorial.html