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

pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy