Repository files navigation

This is a PyTorch Re-Implementation ofEAST: An Efficient and Accurate Scene Text Detector.

• Only RBOX part is implemented.
• Using dice loss instead of class-balanced cross-entropy loss. Some codes refer toargman/EAST andsongdejia/EAST
• The pre-trained model provided achieves82.79 F-score on ICDAR 2015 Challenge 4 using only the 1000 images. seehere for the detailed results.

Only tested on

• Anaconda3
• Python 3.7.1
• PyTorch 1.0.1
• Shapely 1.6.4
• opencv-python 4.0.0.21
• lanms 1.0.2

When running the script, if some module is not installed you will see a notification and installation instructions.if you failed to install lanms, please update gcc and binutils. The update under conda environment is:

conda install -c omgarcia gcc-6conda install -c conda-forge binutils

The original lanms code has a bug innormalize_poly that the ref vertices are not fixed when looping the p's ordering to calculate the minimum distance. We fixed this bug inLANMS so that anyone could compile the correct lanms. However, this repo still uses the original lanms.

git clone https://github.com/SakuraRiven/EAST.gitcd EAST

• Download Train and Test Data:ICDAR 2015 Challenge 4. Cut the data into four parts: train_img, train_gt, test_img, test_gt.

• Download pre-trained VGG16 from PyTorch:VGG16 and our trained EAST model:EAST. Make a new folderpths and put the download pths intopths

mkdir pthsmv east_vgg16.pth vgg16_bn-6c64b313.pth pths/

Here is an example:

.├── EAST│   ├── evaluate│   └── pths└── ICDAR_2015    ├── test_gt    ├── test_img    ├── train_gt    └── train_img

Modify the parameters intrain.py and run:

CUDA_VISIBLE_DEVICES=0,1 python train.py

Modify the parameters indetect.py and run:

CUDA_VISIBLE_DEVICES=0 python detect.py

• The evaluation scripts are fromICDAR Offline evaluation and have been modified to run successfully with Python 3.7.1.
• Change theevaluate/gt.zip if you test on other datasets.
• Modify the parameters ineval.py and run:

CUDA_VISIBLE_DEVICES=0 python eval.py