Skip to content

Classify MNIST image dataset into 10 classes. Build an image classifier with Recurrent Neural Network (RNN: LSTM) on Tensorflow.

License

Notifications You must be signed in to change notification settings

jiegzhan/image-classification-rnn

Repository files navigation

Project: Build an Image Classifier with RNN(LSTM) on Tensorflow

Highlights:

  • This is a multi-class image classification problem.
  • The purpose of this project is to classify MNIST image dataset into 10 classes.
  • The model was built with Recurrent Neural Network (RNN: LSTM) on Tensorflow.

Data:

  • The MNIST image dataset was saved in the ./data/ directory.

Train:

  • Command: python3 train.py parameters.json
  • Example: python3 train.py ./parameters.json

A directory will be created during training, and the model will be saved in this directory.

Predict:

Provide the model directory (created when running train.py) to predict.py.

  • Command: python3 predict.py ./trained_model_directory/
  • Example: python3 predict.py ./trained_model_1481170507/

Reference:

About

Classify MNIST image dataset into 10 classes. Build an image classifier with Recurrent Neural Network (RNN: LSTM) on Tensorflow.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages