Skip to content

An autoencoder using a convolutional neural network with Tensorflow

Notifications You must be signed in to change notification settings

zbn123/Autoencoder

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

68 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

MNIST Autoencoder

Just an autoencoder example, specifically for the MNIST dataset. This compresses the image from 786 dimmensions down to 8.

Usage

>>> python autoencoder.py

This will automatically download the MNIST dataset. Batch size is set to 1000 so you might want to change it depending on your system. A checkpoint is included in the checkpoint folder, and will automatically be loaded upon running. To train your own, delete the checkpoint.

Results

It didn't do too bad considering the images are compressed to only 8 dimensions. Below are some examples.

img img

img img

img img

img img

About

An autoencoder using a convolutional neural network with Tensorflow

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 99.5%
  • Shell 0.5%