Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

AI_NumberGuessing

CNN based torch neural network for guessing number from 0 to 9.

Dataset

MNIST dataset with self-implemented data augumentation

Model

PyTorch CNN model, trained on 10k images

  • accuracy on test 97.6

Application

Bacis application on PyGame for testing model

How to use

  • clone this repo
  • run main.py
  • or run output/main.exe

Examples

example_digit3 example_digit9 example_digit4

About

CNN based torch neural network

Resources

Stars

Watchers

Forks

Releases

Packages

Contributors

Languages