Skip to content

Deep Convolutional Generative Adversarial Network (DCGAN) implementation using PyTorch trained on the MNIST dataset to generate images of handwritten digits

Notifications You must be signed in to change notification settings

OmarAzizi/DCGAN-Implementation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 

Repository files navigation

Deep Convolutional Generative Adversarial Network (DCGAN)

PyTorch

DCGAN is one of the popular and successful network designs for GAN. It's mainly composed of convolution layers without max pooling or fully connected layers. It uses convolutional stride and transposed convolution for the downsampling and the upsampling. The figure below is the network design for the generator.

image

After adding the Discriminator we will have a network that roughly looks like this:

image

Performance

This is a short video for the model running for 5 epochs on MNIST dataset.

Screencast.from.2024-03-16.01-40-22.mp4

References

This implementation for DCGAN is from this research paper

Unsupervised Representation Learning With Deep Convolutional Generative Adversarial Networks

About

Deep Convolutional Generative Adversarial Network (DCGAN) implementation using PyTorch trained on the MNIST dataset to generate images of handwritten digits

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages