Skip to content

Convolutional recurrent network in pytorch

License

Notifications You must be signed in to change notification settings

youscan/crnn.pytorch

 
 

Repository files navigation

Convolutional Recurrent Neural Network

This software implements the Convolutional Recurrent Neural Network (CRNN) in pytorch. Origin software could be found in crnn

Train a new model

  1. Construct dataset and alphabeth following generate_lmdb_dataset.ipynb
  2. Execute PYTHONIOENCODING=utf-8 python3.6 train.py --adadelta --trainRoot train_sample --valRoot test_sample --cuda --random_sample -alphabeth alphabeth. Explore train.py for details.

Run demo

Put the downloaded model file crnn.pth into directory data/. Then launch the demo by:

python demo.py

The demo reads an example image and recognizes its text content.

Example image: Example Image

Expected output: loading pretrained model from ./data/crnn.pth a-----v--a-i-l-a-bb-l-ee-- => available

About

Convolutional recurrent network in pytorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 78.2%
  • Jupyter Notebook 15.9%
  • Lua 5.9%