yanhongyu111 / crnn.pytorch

A pytorch re-implementation of Convolutional recurrent network in pytorch

Convolutional Recurrent Neural Network

Requirements

• python 3.5+
• pytorch 1.1
• 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 and copy the output to alphabet in config.json

python3 utils/get_keys.py --label_file /path/to/label_file1 /path/to/label_file2 ...

2. use fellow 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 fellow script to predict

python3 predict.py