/MNIST-invert-color

Invert the color of MNIST images with PyTorch (2017)

Primary LanguagePython

MNIST Invert Color

Inverting the color of MNIST images from black to white and white to black using CycleGAN implemented with PyTorch.


Prerequites

  • Python 3.6
  • PyTorch 0.1.12 (PyTorch is currently available only on Linux and OSX)
  • The code has been written on Linux (ubuntu 16.04 LTS) system

How to

1. Install Python and PyTorch (from the link above or on your own way)

2. Run the command below on Terminal

$ python CycleGAN_MNIST_Invert.py

Results

1) Black Background -> White Background

Number of Iteration Original Image Generated (Inverted) Image
0 alt text alt text
15 alt text alt text
300 alt text alt text

2) White Background -> Black Background

Number of Iteration Original Image Generated (Inverted) Image
0 alt text alt text
15 alt text alt text
300 alt text alt text