Skip to content

This repository is a simple Pytorch Implementation of DCGAN for generation of human faces using CelebA dataset

Notifications You must be signed in to change notification settings

partha2409/DCGAN_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DCGAN_pytorch

This repository is a simple Pytorch Implementation of DCGAN https://arxiv.org/abs/1511.06434 for generation of human faces using CelebA dataset.

Requriements: Pytorch, PIL, Matplotlib, Pickle

Dataset: Download CelebA images from https://www.kaggle.com/jessicali9530/celeba-dataset

To Train the model:

  1. Update the dataset path 'train_data_path' in config.py
  2. Run Train.py

To Generate images:

  1. Update 'model_dir and 'model_path' of the trained model in 'run_inference.py'
  2. Run run_inference.py

About

This repository is a simple Pytorch Implementation of DCGAN for generation of human faces using CelebA dataset

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages