Skip to content

I trained a PyTorch CNN on the MNIST dataset and added a pygame canvas, where the user can draw digits and the model tries to recognize them

Notifications You must be signed in to change notification settings

b4sen/digit_recognizer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Digit Recognizer

I created a very basic CNN in PyTorch and trained the model on the MNIST dataset using Google Colabs GPU. The user can draw a numerical digit on a pygame canvas and the model tries to recognize it. This project is still WIP, but the basic functionalities are working.

use the digit_recognizer.py as entrypoint.

this model has achieved a score of 0.99675 on the kaggle Digit Recognizer challenge

About

I trained a PyTorch CNN on the MNIST dataset and added a pygame canvas, where the user can draw digits and the model tries to recognize them

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published