A simple digits recognition neural network
Project description
# Convolutional Neural Network using MNIST dataset for Digit Recognition
Official repository: https://github.com/MartinBraquet/ml-digits-recognition.
Test online: https://martinbraquet.com/index.php/solo_page_digits_recognition.
![Alt Text](https://github.com/MartinBraquet/ml-digits-recognition/blob/main/src/demo.gif)
## Installation
` pip install -r requirements.txt `
## Documentation
Convolutional neural network visualization.
` nn_visualization.ipynb `
![Alt Text](https://github.com/MartinBraquet/ml-digits-recognition/blob/main/src/nn1.png)
![Alt Text](https://github.com/MartinBraquet/ml-digits-recognition/blob/main/src/nn2.png)
## Training
Train the model and save it as model.pt.
` ml_digits_recognition_training.ipynb `
Accuracy vs epochs.
![Alt Text](https://github.com/MartinBraquet/ml-digits-recognition/blob/main/src/accuracy.png)
Loss vs epochs.
![Alt Text](https://github.com/MartinBraquet/ml-digits-recognition/blob/main/src/loss.png)
## Test
Test in Jupiter Notebook. The model can be loaded from the training above in model.pt or from the default precise model in model_precise.pt.
` ml_digits_recognition_test.ipynb `
Test in Python.
` python user_input_drawing.py `
## Tools
Draw a digit and save it as a PNG file.
` user_input_drawing.ipynb `
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for ml_digits_recognition-0.0.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 05f7121a6aac32c8a19b028e4129a3899b23cd894125955cd63195c97855d7f5 |
|
MD5 | 25f433894aaf1416cc6aac5618c33cdf |
|
BLAKE2b-256 | 12d8391e07a53e8e31bb8600f1b5e6d83661f9491e4687ead7928f84d9748c36 |
Hashes for ml_digits_recognition-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | e0496169b2248be0a3a3f0b0cc8be98db95882bee2004d4cddb5423ec0afc3a7 |
|
MD5 | dd8a314aff7012676b79e02638516276 |
|
BLAKE2b-256 | 21ba41e17ced8d151725a261e87fe74fafc4907c093de29fd902972cef52a578 |