Pytorch implements Real-time Scene Text Detection with Differentiable Binarization
- Python 2.7
- PyTorch v0.4.1+
- pyclipper
- Polygon2
- OpenCV 4.0
- opencv-python 3.4
python train_ic15.py --arch resnet50 --batch_size 8 --root_dir ../../../Store/Text-Detection-Datasets/ICDAR2017-MLT/train/
python eval.py --root_dir ../../../Store/Text-Detection-Datasets/ICDAR2017-MLT/val/ --resume checkpoints/ic15_resnet50_bs_8_ep_600/checkpoint.pth.tar --long_size 1280
Probability map | Threshold map | binarization map | Image |
---|---|---|---|
![]() |
![]() |
![]() |
![]() |
PSENet Real-time Scene Text Detection with Differentiable Binarization