Skip to content

2abet/Fashion-MNIST-GANS

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Fashion Image Generation with GANs

Project Overview

This project involves using Generative Adversarial Networks (GANs) to generate fashion images. It employs TensorFlow to build and train a GAN model on the Fashion MNIST dataset. The project showcases the construction of both the generator and discriminator models, training these models, and then using the trained generator to produce new, synthetic fashion images.

Prerequisites

Before you begin, ensure you meet the following requirements:

  • Python 3.8 or higher
  • TensorFlow 2.x
  • TensorFlow Datasets
  • Matplotlib
  • Numpy

Installation

Follow these steps to get your development environment running:

  1. Clone the repository:
    git clone <repository-url>
  2. Install the required Python packages:
    pip install tensorflow tensorflow_datasets matplotlib numpy

Usage

To use this project, follow these steps:

  1. Load the dataset and preprocess it:

    • Scale the images to a range between 0 and 1.
    • Shuffle and batch the dataset for training.
  2. Build the neural network models:

    • Construct the generator and discriminator models using TensorFlow's Sequential API.
  3. Compile and train the GAN:

    • Set up loss functions and optimizers for both the generator and discriminator.
    • Train the model using custom training loops.
    • Monitor the training process and save generated images after each epoch.
  4. Generate fashion images:

    • Use the trained generator model to produce fashion images.
  5. Save and load the generator model for future use.

Example Generated Images

After training, the generator will be capable of producing images like the ones shown below (example images should be included in the project repository under an images/ directory).

Acknowledgments

This project was inspired by the TensorFlow and Keras documentation, and the Fashion MNIST dataset provided by Zalando Research.

Contact Information

For help or questions about this project, please contact [Arabambi Akinyemi] at [yemiarabambi@gmail.com].

About

Fashion Items Generator

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages