Skip to content

Generative Adversarial Network (GAN) built with CNNs for both the generator and discriminator, implemented in TensorFlow and trained on the Fashion MNIST dataset.

Notifications You must be signed in to change notification settings

Szymon-Budziak/Fashion_MNIST_GAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Fashion MNIST Generative Adversarial Network

This project is a simple implementation of a Generative Adversarial Network (GAN) using the Fashion MNIST dataset. The GAN is implemented using Tensorflow and trained on the Fashion MNIST dataset. The GAN is trained to generate new images of clothing items that are similar to the images in the Fashion MNIST dataset.

Installation

The requirements of the project are listed in the requirements.txt file. To install the requirements, run the following command:

  1. Create a virtual environment and activate it:
python -m venv venv
source venv/bin/activate
  1. Install the requirements:
pip install -r requirements.txt

Usage

All the code for the project is in the main.ipynb notebook. Architectures of the generator and discriminator are located in separate directories called generator and discriminator respectively. Code for training the GAN is in the notebook.

Results

The GAN was trained for 10000 epochs on the Fashion MNIST dataset with 32 batch size. The results of the training are:

  • After 500 epochs

500 epochs

  • After 1000 epochs

1000 epochs

  • After 5000 epochs

5000 epochs

  • After 9500 epochs

10000 epochs

About

Generative Adversarial Network (GAN) built with CNNs for both the generator and discriminator, implemented in TensorFlow and trained on the Fashion MNIST dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published