omar-mohamed / Home-Number-Recognition-SVHN

A deep learning project to recognize home numbers in an image.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Home-Number-Recognition-SVHN

A deep learning project to recognize home numbers in an image.

Problem:

Given an image, the task is to identify the number in the image. The number to be identified is a sequence of digits, s = s1; s2; : : : ; sn. When determining the accuracy of a digit transcriber, we compute the proportion of the input images for which the length n of the sequence and every element si of the sequence is predicted correctly.

g1

Dataset:

SVHN is a real-world image dataset for developing machine learning and object recognition algorithms with minimal requirement on data preprocessing and formatting. SVHN is obtained from house numbers in Google Street View images.

Link: SVHN_Dataset

Dataset Splitting:

  • The original SVHN dataset contains around 33k training image and around 13k test image.
  • Download extra training examples from SVHN website around(202k images)
  • Add 150k image to training, 50k to test, and around 2300 to validation set.
  • Now the training set is around 183k picture
  • The test set is around 63k picture
  • The validation set is around 2300 picture

Preprocessing:

  • Cropping using the bounding boxes provided in the dataset to 32x32
  • Grayscale and Normalization

g2

Training Architecture:

g3

For more details on the architecture please check our presentation slides.

Results:

  • Reached 89.4% on the 63k test images.
  • Reached 91% on 2300 images validation set.

g4

g5

GUI Interface:

We made a GUI interface that takes a picture and classify it using our saved model.

g6

Future Work:

  • Try different architectures, meta-parameters, and loss functions to increase accuracy.
  • Try to add numbers localization so it can detect numbers regardless of zoom or scale.

References:

Environment Used:

  • Python 3.6.1
  • Tensorflow 1.3

About

A deep learning project to recognize home numbers in an image.


Languages

Language:Python 100.0%