crazysummerday / ctpn.pytorch

Pytorch implementation of CTPN (Detecting Text in Natural Image with Connectionist Text Proposal Network)

ctpn.pytorch

Pytorch implementation of CTPN (Detecting Text in Natural Image with Connectionist Text Proposal Network)

Paper

https://arxiv.org/pdf/1609.03605.pdf

train

training dataset: ICDAR2013 and ICDAR2017.
If you want to train your own dataset, you need to change the 'img_dir' and 'label_dir' in file ctpn/config.py, then run

python train.py

predict

Download pretrained model from './weights/', change the test image path in file predict.py, then run:

python predict.py

result

references

https://github.com/opconty/pytorch_ctpn

https://github.com/courao/ocr.pytorch

ctpn.pytorch's Issues

Format of training dataset

Hi, your repo is very useful for me. I have a questions. Can you describe the format of training dataset? How look like structure of dataset prepare? The labels is .xml? I need train my own dataset, wait for your reply

训练数据链接

能不能把训练数据链接贴一下，我找了好几个ICDAR 2017的链接，感觉都不对，谢谢！

about ICDARDataset and VOCdataset

dataset

Thank you very much for sharing your code, can you provide the dataset?

关于给出的已训练好的模型参数

想问一下作者给出的模型参数使用什么数据集训练的，准确率大概是多少~

Text Box

If i want to take some operations in the Text box For example, do Binarization in the text box
What should I do?