YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
Generative Adversarial Networks (GAN) โ Fashion MNIST
This project implements Generative Adversarial Networks (GANs) for image generation using the Fashion-MNIST dataset. We compare two architectures:
- Vanilla GAN (Fully Connected Networks)
- Deep Convolutional GAN (DCGAN)
The goal is to analyze:
- Image quality
- Training stability
- Effect of architecture on performance
Dataset
- Fashion-MNIST
- 28ร28 grayscale images of clothing items
- Dataset automatically downloaded using PyTorch
Preprocessing
- Converted images to tensors
- Normalized pixel values to range [-1, 1]
Model Architectures
1. Vanilla GAN
Generator
- Fully connected layers
- ReLU activations
Discriminator
- Fully connected layers
- LeakyReLU activations
2. DCGAN
Generator
- Transposed Convolution layers
- Batch Normalization
- ReLU activations
Discriminator
- Convolution layers
- Batch Normalization
- LeakyReLU activations
Training Process
Train discriminator on:
- Real images (label = 1)
- Fake images (label = 0)
Train generator to:
- Fool discriminator (label = 1)
Loss function:
- Binary Cross Entropy (BCE)
Optimizer:
- Adam optimizer (lr = 0.0002)
Observations
- DCGAN produces sharper and more realistic images
- Vanilla GAN generates blurry outputs
- Training is unstable in early epochs
Loss Behavior
- Generator and discriminator losses oscillate
- DCGAN shows relatively stable convergence
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐ Ask for provider support