Skip to content

Latest commit

 

History

History
executable file
·
91 lines (67 loc) · 6.41 KB

algorithm_overview_en.md

File metadata and controls

executable file
·
91 lines (67 loc) · 6.41 KB

Two-stage Algorithm

1. Algorithm Introduction

This tutorial lists the text detection algorithms and text recognition algorithms supported by PaddleOCR, as well as the models and metrics of each algorithm on English public datasets. It is mainly used for algorithm introduction and algorithm performance comparison. For more models on other datasets including Chinese, please refer to PP-OCR v2.0 models list.

1.1 Text Detection Algorithm

PaddleOCR open source text detection algorithms list:

On the ICDAR2015 dataset, the text detection result is as follows:

Model Backbone Precision Recall Hmean Download link
EAST ResNet50_vd 88.71% 81.36% 84.88% trained model
EAST MobileNetV3 78.2% 79.1% 78.65% trained model
DB ResNet50_vd 86.41% 78.72% 82.38% trained model
DB MobileNetV3 77.29% 73.08% 75.12% trained model
SAST ResNet50_vd 91.39% 83.77% 87.42% trained model
PSE ResNet50_vd 85.81% 79.53% 82.55% trianed model
PSE MobileNetV3 82.20% 70.48% 75.89% trianed model

On Total-Text dataset, the text detection result is as follows:

Model Backbone Precision Recall Hmean Download link
SAST ResNet50_vd 89.63% 78.44% 83.66% trained model

Note: Additional data, like icdar2013, icdar2017, COCO-Text, ArT, was added to the model training of SAST. Download English public dataset in organized format used by PaddleOCR from:

For the training guide and use of PaddleOCR text detection algorithms, please refer to the document Text detection model training/evaluation/prediction

1.2 Text Recognition Algorithm

PaddleOCR open-source text recognition algorithms list:

Refer to DTRB, the training and evaluation result of these above text recognition (using MJSynth and SynthText for training, evaluate on IIIT, SVT, IC03, IC13, IC15, SVTP, CUTE) is as follow:

Model Backbone Avg Accuracy Module combination Download link
Rosetta Resnet34_vd 80.9% rec_r34_vd_none_none_ctc trained model
Rosetta MobileNetV3 78.05% rec_mv3_none_none_ctc trained model
CRNN Resnet34_vd 82.76% rec_r34_vd_none_bilstm_ctc trained model
CRNN MobileNetV3 79.97% rec_mv3_none_bilstm_ctc trained model
StarNet Resnet34_vd 84.44% rec_r34_vd_tps_bilstm_ctc trained model
StarNet MobileNetV3 81.42% rec_mv3_tps_bilstm_ctc trained model
RARE MobileNetV3 82.5% rec_mv3_tps_bilstm_att trained model
RARE Resnet34_vd 83.6% rec_r34_vd_tps_bilstm_att trained model
SRN Resnet50_vd_fpn 88.52% rec_r50fpn_vd_none_srn trained model
NRTR NRTR_MTB 84.3% rec_mtb_nrtr trained model
SAR Resnet31 87.2% rec_r31_sar trained model
SEED Aster_Resnet 85.2% rec_resnet_stn_bilstm_att trained model

Please refer to the document for training guide and use of PaddleOCR

2. Training

For the training guide and use of PaddleOCR text detection algorithms, please refer to the document Text detection model training/evaluation/prediction. For text recognition algorithms, please refer to Text recognition model training/evaluation/prediction

3. Inference

Except for the PP-OCR series models of the above models, the other models only support inference based on the Python engine. For details, please refer to Inference based on Python prediction engine