This repository contains a Convolutional Neural Network (CNN) model trained to recognize handwritten digit images. The model is capable of accurately classifying digits from 0 to 9. This project serves as a demonstration of implementing deep learning techniques for image classification tasks.
The model was trained on the popular MNIST dataset, which consists of 28x28 grayscale images of handwritten digits.
-
Conv2D layer with 32 filters, 5x5 kernel size, 'Same' padding, and 'relu' activation
-
Conv2D layer with 32 filters, 5x5 kernel size, 'Same' padding, and 'relu' activation
-
MaxPooling2D layer with 2x2 pool size
-
Dropout layer with a dropout rate of 0.25
-
Conv2D layer with 64 filters, 3x3 kernel size, 'Same' padding, and 'relu' activation
-
Conv2D layer with 64 filters, 3x3 kernel size, 'Same' padding, and 'relu' activation
-
MaxPooling2D layer with 2x2 pool size and 2x2 strides
-
Dropout layer with a dropout rate of 0.25
-
Flatten layer
-
Dense layer with 256 units and 'relu' activation
-
Dropout layer with a dropout rate of 0.5
-
Output layer with 10 units (for 10 digits) and 'softmax' activation
The model was trained using TensorFlow and Keras on a GPU-enabled machine. The training process involved 2 epochs and a batch size of 80. The model achieved an accuracy of [98]% on the validation dataset.