Skip to content

bb-in-hoodie/MNIST-invert-color

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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

Releases

No releases published

Packages

No packages published

Languages