Skip to content

This project is a PyTorch implementation that uses deep CNN to recognize multi-digit numbers using the SVHN dataset derived from Google Street View house numbers, each picture contains a set of numbers from 0 to 9, the model is tested to have 89% accuracy.| 使用深度卷积神经网络从街景图像中识别多位数门牌号的PyTorch实现方案,使用的数据集为SVHN,来源于谷歌街景门牌号码,每张图片中包含一组0-9的阿拉伯数字,经测试精确度可达89%

License

Notifications You must be signed in to change notification settings

MuGeminorum/SVHN-Recognition

Repository files navigation

SVHN-Recognition

license Python application hf ms

This project is a PyTorch implementation that uses deep CNN to recognize multi-digit numbers using the SVHN dataset derived from Google Street View house numbers, each picture contains a set of numbers from 0 to 9, the model is tested to have 89% accuracy.

Environment

conda create -n svhn --yes --file conda.txt
conda activate svhn
pip install -r requirements.txt

Usage

  1. Clone the source code:
git clone git@github.com:MuGeminorum/SVHN-Recognition.git
cd SVHN-Recognition
  1. Run convert_to_lmdb.py
  2. Run train.py

Params

Steps GPU Batch Size Learning Rate Patience Decay Step Decay Rate Accuracy
122000 GTX 1080 Ti 512 0.01 100 625 0.9 89.21%

Training curve

Reference

[1] Multi-digit Number Recognition from Street View Imagery using Deep Convolutional Neural Networks

About

This project is a PyTorch implementation that uses deep CNN to recognize multi-digit numbers using the SVHN dataset derived from Google Street View house numbers, each picture contains a set of numbers from 0 to 9, the model is tested to have 89% accuracy.| 使用深度卷积神经网络从街景图像中识别多位数门牌号的PyTorch实现方案,使用的数据集为SVHN,来源于谷歌街景门牌号码,每张图片中包含一组0-9的阿拉伯数字,经测试精确度可达89%

Topics

Resources

License

Stars

Watchers

Forks

Languages