This repository contains code for training and evaluating a digit recognition model using the MNIST dataset.
- Python 3.x
- TensorFlow 2.x
- NumPy
- Matplotlib
Clone the repository:
git clone https://github.com/oxyboron25/mnist-digit-recognition.git
cd mnist-digit-recognition
Install dependencies using pip:
pip install -r requirements.txt
To train the model, run:
python train.py
This will train the digit recognition model using the MNIST dataset.
To evaluate the model on the test set, run:
python evaluate.py
This will evaluate the accuracy of the trained model on the MNIST test dataset.
To predict digits from custom images, use:
python predict.py path_to_image
Replace path_to_image
with the path to the image file you want to predict.
data/
: Contains the MNIST dataset (automatically downloaded if not present).models/
: Saved models after training.train.py
: Script for training the digit recognition model.evaluate.py
: Script for evaluating the model's accuracy.predict.py
: Script for predicting digits from custom images.