CoolFace
Apppublic

ju4nppp/dcgan-mnist-demo

sourceHugging Faceupdated 1y agoView on Hugging Face
0likes
App README

DCGAN MNIST Generator

This repository contains a Deep Convolutional GAN (DCGAN) trained on the MNIST dataset. The model generates handwritten-like digit images from random noise.

Model Architecture

The model implementation is based on the paper Unsupervised Representation Learning with Deep Convolutional Generative Adversarial Networks.

  • —Generator architecture: 5 transposed convolutional layers with batch normalization
  • —Latent space dimension: 100
  • —Output: 64x64 grayscale images

Demo App

The included Gradio app allows you to generate new MNIST-like images using the pre-trained model.

Running Locally

  1. 1.Install dependencies:
bash
pip install -r requirements.txt