identify-numerical-digits
Created 4 years ago
Maintained by
srikavin
A neural net that identifies digits trained using the MNIST dataset
Watch
1
Members
1
Srikavin Ramkumar committed 4 years ago
GCI - Identify Numerical Digits
This repository contains code for training a neural network in pytorch to classify MNIST digits.
Dependencies
The following dependencies are used: PyTorch PyTorch Vision Numpy Matplotlib
They can be installed by running pip install -r requirements.txt.
Usage
identify.py is a CLI application used to identify digits in an image.
main.ipynb is a Jupyter notebook that can be used to train the model using the MNIST dataset.
It can also be viewed on Google Colab.
usage: identify.py [-h] [-m MODEL] input_file
Example
There are two test images in this repository. The trained model can be used to identify the numbers within the images using the following commands:
python identify.py test.png
python identify.py test1.png
A screen recording is available on YouTube.