Skip to content

Application where user draws numbers and convolutional neural network predicts the number

License

Notifications You must be signed in to change notification settings

Jarsse/Draw_Number_CNN_Predicts

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Draw number and CNN predicts it

Application where user draws numbers and convolutional neural network predicts the number. savedModels contains the current CNN model. The model had validation accuracy of 99.017 with 47 training epochs. Model contains convolutional layer (16 filters, 3x3 kernel), max pooling with 2x2 pool size, 2x convolutional layers (32 filter 3x3 kernel, 64 filter, 3x3 kernel), another max pooling with 2x2 pool size, flatten, 2 dense layers with 32 units and relu activation and finally dense layer with 10 units and softmax layer.

The training was done with the MNIST_Training.py. The main program is in main.py and mnistapp.kv files.

The program starts when the user presses the "Start" button. Then the user can draw on the black area. start The "Reset" button removes the current drawings and predict button asks the neural network to predict which number is currently drawn. Currently the program also has 20% chance to give prediction each time the user draws something on the screen. two The network predicts the numbers pretty well. three

five Sometimes the network gives wrong answers but so far it has given the right prediction after more precise drawing or whitening the image with more clicks.

About

Application where user draws numbers and convolutional neural network predicts the number

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages