Skip to content

mitya52/attention-ocr

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Attention OCR

This folder contains simple tensorflow implementation of attention ocr model from paper Attention-based Extraction of Structured Information from Street View Imagery.

Requirements

You need to install the TensorFlow, OpenCV and NumPy libraries.

Getting Started

Show data:

./generator.py

Train model on simple data, generated by generator.py:

./train.py

Test model with visualizing attention weights on each decoding step (use pretrained model from train folder):

./predict.py

Example of predict.py output:

Authors

Dimitry Ageev - mitya52

About

Attention OCR

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages