Convolutional Recurrent Neural Network (CRNN) Implement with pytorch for chinese OCR.
- Prepare training data, you can find samples(with labels) in the folder of "data", it looks like:
-
Create LMDB database cd create_dataset python create_dataset.py
-
train CRNN for chinese OCR cd pytorch-train python crnn_main.py --cuda
you can find trained model in the folder of "save_model"
notes:
- needs to be carefully confirmed whether your Chinese characters are in alphabet or not.
- good performance in training accuracy, and bad performance in practical application? maybe samples of various situations are needed.