Modified codes from pytorch_ctpn
Add OHEM
Support ICDAR17MLT dataset
To train your own model, put your images into one directory [images],
and labels into another directory [labels].
Replace the value of icdar17_mlt_img_dir and icdar17_mlt_gt_dir in config.py by your own path.
Then run
python3 ctpn_train.py
If you want to train on ICDAR datasets, please visit here and download datasets you like.