/crnn.pytorch

A pytorch re-implementation of Convolutional recurrent network in pytorch

Primary LanguagePythonMIT LicenseMIT

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