Skip to content

A pytorch re-implementation of Convolutional recurrent network in pytorch

License

Notifications You must be signed in to change notification settings

NextGuido/crnn.pytorch

 
 

Repository files navigation

Convolutional Recurrent Neural Network

Requirements

  • python 3.5+
  • pytorch 1.1
  • torch_baidu_ctc
  • lmdn
  • opencv

Data Preparation

Prepare a text in the following format

/path/to/img/img.jpg helloword
...

Train

  1. config the train_file,test_filein config.json
  2. generate alphabet use fellow script to generate alphabet.py in the some folder with train.py
python3 utils/get_keys.py
  1. use following script to run
python3 train.py

Predict

predict.py is used to inference on single image

  1. config model_path, img_path in predict.py
  2. use following script to predict
python3 predict.py

About

A pytorch re-implementation of Convolutional recurrent network in pytorch

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 73.2%
  • Jupyter Notebook 26.8%