magnetstone / pytorch_ctpn Goto Github PK

update 19-03-20 wed: android ocr

This is a pytorch implementation of CTPN(Detecting Text in Natural Image with Connectionist Text Proposal Network).Inspired by keras-ocr.

Training log is available:Training Log(Chinese)

• download ctpn model weights (converted from keras ctpn weights) ctpn_keras_weights.pth.tar from dropbox, and move it to ./checkpoints/ (For a number of reasons, the pretrained weights will no longer be available.Thanks for your attention.)
• download VOC2007_text_detection Chinese Text Detection dataset and move it to ./images/
• run python ctpn_train.py --image-dir image_dir --labels-dir labels_dir --num-workers num_workers

• download the pretrained weights from dropbox
• Please refer to predict.py for more details.

Training Log(Chinese)

These days, I'm working on deploying this model on Android devices.you can check the results from here.

Android text recognition 4-23

Find out that adopting skew transform can significantly improve recognition accuracy.(It may take a few seconds, heavily depends on your harware and input image size)

• CTPN (Detecting Text in Natural Image with Connectionist Text Proposal Network)
• keras-ocr

MIT License