Skip to content

neemiasbsilva/autoencoder-mnits

Repository files navigation

Building Auto Encoder Compression and Use to Mnist Dataset

This repository was created to build the deep Convolutional Neural Networks autoencoder to apply on mnist dataset.

To know more about the implementation, bellow follows a brief description about the files:

  • Model.py: this files contains the building of cnn architecture and the auto encoder construction;
  • Generator Dataset: get the mnist dataset;
  • The main file: this file contains main code that use to fit the auto encoder compression;
  • Visualize: this file contains the result of auto encoder compression, which that run as the result the auto encoder compression.

Note: For run this codes is necessary create a virtual environment and install the tensorflow, keras, and matplotlib.

Sincerely: Neemias B. Silva