Pytorch CTPN
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)
| model | size |
|---|---|
| keras-CTPN | 142M |
| pytorch-CTPN | 67.6M |
train
download ctpn model weights (converted from keras ctpn weights)(For a number of reasons, the pretrained weights will no longer be available.Thanks for your attention.)ctpn_keras_weights.pth.tarfrom dropbox, and move it to ./checkpoints/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
predict
download the pretrained weights from dropbox- Please refer to predict.py for more details.
results
Training Log(Chinese)
Android DEMO
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)
