Skip to content

yusufM03/Detect-Digits-using-CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Digit Recognition using Convolutional Neural Networks (CNN)

Digit Recognition

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.

Dataset

The model was trained on the popular MNIST dataset, which consists of 28x28 grayscale images of handwritten digits.

Model Architecture

The CNN model architecture consists of the following layers: Model Architecture:

  1. Conv2D layer with 32 filters, 5x5 kernel size, 'Same' padding, and 'relu' activation

  2. Conv2D layer with 32 filters, 5x5 kernel size, 'Same' padding, and 'relu' activation

  3. MaxPooling2D layer with 2x2 pool size

  4. Dropout layer with a dropout rate of 0.25

  5. Conv2D layer with 64 filters, 3x3 kernel size, 'Same' padding, and 'relu' activation

  6. Conv2D layer with 64 filters, 3x3 kernel size, 'Same' padding, and 'relu' activation

  7. MaxPooling2D layer with 2x2 pool size and 2x2 strides

  8. Dropout layer with a dropout rate of 0.25

  9. Flatten layer

  10. Dense layer with 256 units and 'relu' activation

  11. Dropout layer with a dropout rate of 0.5

  12. Output layer with 10 units (for 10 digits) and 'softmax' activation

Training

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.

About

Detect digits using CNN /Avoid overfitting -->Data augmentation/Dropout/Transfer learning

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published