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 |
- download ctpn model weights (converted from keras ctpn weights)
ctpn_keras_weights.pth.tar
from 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
- download the pretrained weights from dropbox
- Please refer to predict.py for in more detail
Training Log(Chinese)