Skip to content

HoangLayor/Handwritten-Digit-Recognition

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

53 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Handwritten Digit Recognition using PyTorch

This project trains a simple DL model on the MNIST dataset to recognize handwritten digits.

Installation

pip install -r requirements.txt

Training & Evaluation

📂 src
├── 📂 checkpoints
│   ├── 📂 checkpoint_1
│   └── 📂 checkpoint_2
├── 📂 data
│   ├── 📂 MNIST
│   └── 📄 dataset.py
├── 📂 models
│   └── 📄 model.py
├── 📂 test_images
│   ├── 📄 00.png
│   ├── 📄 01.png
│   └── ...
├── 📂 training
│   ├── 📄 evaluation.py
│   └── 📄 train.py
├── 📂 utils
│   └── 📄 utils.py
├── 📄 main.py
└── 📄 mnist.ipynb

Train & Eval

python src/main.py

Deploy

API

📂 api_app
├── 📂 models
│   └── 📄 model.py
├── 📂 utils
│   ├── 📄 preprocess.py
│   └── 📄 utils.py
├── 📄 api.py
├── 📄 inference.py
└── 📄 run.py

Run FastAPI

python api_app/run.py

Streamlit

📂 streamlit_app
├── 📂 models
│   └── 📄 model.py
├── 📂 utils
│   ├── 📄 predict.py
│   ├── 📄 preprocess.py
│   └── 📄 utils.py
└── 📄 streamlit_app.py

Run streamlit app

streamlit run streamlit_app/streamlit_app.py

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •