Detecting digit on the images using Convolution Neural Networks
-
The neural network network is trained on the grayscale images provided by kaggle form MNIST competition. "Data" folder contains the data to train the images.
-
"conv_net.py" script builds the convolutional network and dumps it in .hdf5 file, which reduces the overhead of training the model again.
-
Flask App folder contains the code for the app implmented here
- 2D Convoultion
- Relu Activation
- 2D MAX Pooling
- 2D Convoultion
- Relu Activation
- 2D MAX Pooling
- Dense layer with ouput dim of 500
- Relu Activation
- Ouput Layer with dim of 10
- Softmax Activation
- Tensorflow
- Keras
- Theano
- h5py
- Flask