xiesibo / pytorch_ctpn

This is a pytorch implementation of CTPN(Detecting Text in Natural Image with Connectionist Text Proposal Network)

Pytorch CTPN

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) 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

predict

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

results

Training Log(Chinese)

reference

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

Licence

MIT License