rabbit666666 / pytorch-text-recognition

Text recognition combo - CRAFT + CRNN

Text recognition tool

Text recognition with Pytorch Using CRNN and CRAFT pretrained models.

Main sources:

• Character Region Awareness for Text Detection Paper
• [CRNN Paper]

Install it, install requirements.txt by

python -m pip install requirements.txt

and check argparsers. at the moment it is only input image so

> source venv/Scripts/activate
> python app.py --input_file test_image.jpeg

Example results

return .json file with

{"0": "dollar", "1": "glen", "2": "and", "3": "campbeli", "4": "castle"}�0

and wil save it in default path.

Installing

python setup.py install

Deployment

to do :

add docker/flask/redis to store and visualize results

To dos

Flask app Bootstrap frontend

docker compose

onxx it and put on mobile. how to make it faster? (todo)

Versioning

0.1a

Authors

Acknowledments

Input image (TODO: Preprocess origin format) Output - file with predicted words on input file

Image -> CRAFT -> CRNN -> .json

UPDATE

added simple flask app.