phybrain / BankCard-Recognizer

Extracting numbers from bankcard, based on Deep Learning. 基于深度学习的银行卡号识别与定位系统。

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

BankCard-Recognizer

Extracting numbers from bankcard, based on Deep Learning with Keras.

Including auto and manual location, number identification, with GUI.

中文BLOG: 点击此处链接

bankcard

Roadmap

  • cnn_blstm_ctc
  • EAST/manual locate
  • GUI

Requirements

Python == 3.6

pip install requirements

Environment

Windows10 x64, Anaconda, PyCharm 2018.3, NVIDIA GTX 1050.

Usage

  1. Download trained model, CRNN extracting-code:6eqw, EAST extracting-code:qiw5.
  2. Then put CRNN model into crnn/model, put EAST model into east/model.
  3. Run python demo.py.
  4. In GUI, press Load button to load one image about bankcard or load from dataset/test/.
  5. Press Identify button, it will start locate and do identification.
  6. Activate manual location by double click the Image view, then draw the interest area and press Identify.

Training

Prepare

Download my dataset, CRNN extracting-code:1jax, EAST extracting-code:pqba. and unzip dataset in ./dataset.

The structure of dataset looks like:

- dataset
  - /card          # for east
  - /crad_nbr      # for crnn
  - /test
  ...

for CRNN

  1. Run python crnn/preprocess.py.
  2. Run python crnn/run.py to train, and you can change some parameters in crnn/cfg.py.

for EAST

  1. My dataset is collecting from Internet: Baidu, Google, and thanks Kesci. It has been labeld with ICDAR 2015 format, you can see it in dataset/card/txt/. This tiny dataset is unable to cover all the situation, if you have rich one, it may perform better.
  2. If you would like to get more data, make sure data has been labeled, or you can take dataset/tagger.py to label it.
  3. Modify east/cfg.py, see default values.
  4. Run python east/preprocess.py. If process goes well, you'll see generated data like this:

act_gt

  1. Finally, python east/run.py.

About

cnn_blstm_ctc

The model I used, refer to CNN_RNN_CTC. The CNN part is using VGG, with BLSTM as RNN and CTC loss.

The model's preview:

model

EAST/manual locate

Auto locate is using one famous Text Detection Algorithm - EAST. See more details.

In this project, I prefer to use AdvancedEAST. It is an algorithm used for Scene-Image-Text-Detection, which is primarily based on EAST, and the significant improvement was also made, which make long text predictions more accurate. Original repo see Reference 1.

Also, training process is quiet quick and nice. As practical experience, img_size is better to be 384. The epoch_nbr is no longer important any more, for img_size like 384, usually training will early stop at epoch 20-22. But if you have a large dataset, try to play with these parameters.

This model's preview:

model

Manual locate is only available in GUI. Here're some performance in .gif:

manual-locate1

manual-locate2

GUI

Using QtDesigner to design UI, and PyQt5 to finish other works.

Statement

  1. The bankcard images are collecting from Internet, if any images make you feel uncomfortable, please contact me.
  2. If you have any issues, post it in Issues.

Reference

  1. Advanced EAST - https://github.com/huoyijie/AdvancedEAST
  2. EAST model - https://www.cnblogs.com/lillylin/p/9954981.html

About

Extracting numbers from bankcard, based on Deep Learning. 基于深度学习的银行卡号识别与定位系统。

License:MIT License


Languages

Language:Python 100.0%