2022-08-16 10:46:09 +00:00

5.8 KiB

PP-Structure Model list

1. Layout Analysis

model name description download label_map
ppyolov2_r50vd_dcn_365e_publaynet The layout analysis model trained on the PubLayNet dataset, the model can recognition 5 types of areas such as text, title, table, picture and list inference model / trained model {0: "Text", 1: "Title", 2: "List", 3:"Table", 4:"Figure"}
ppyolov2_r50vd_dcn_365e_tableBank_word The layout analysis model trained on the TableBank Word dataset, the model can only detect tables inference model {0:"Table"}
ppyolov2_r50vd_dcn_365e_tableBank_latex The layout analysis model trained on the TableBank Latex dataset, the model can only detect tables inference model {0:"Table"}

2. OCR and Table Recognition

2.1 OCR

model name description inference model size download
en_ppocr_mobile_v2.0_table_det Text detection model of English table scenes trained on PubTabNet dataset 4.7M inference model / trained model
en_ppocr_mobile_v2.0_table_rec Text recognition model of English table scenes trained on PubTabNet dataset 6.9M inference model / trained model

If you need to use other OCR models, you can download the model in PP-OCR model_list or use the model you trained yourself to configure to det_model_dir, rec_model_dir field.

2.2 Table Recognition

model description inference model size download
en_ppocr_mobile_v2.0_table_structure English table recognition model trained on PubTabNet dataset based on TableRec-RARE 18.6M inference model / trained model
en_ppstructure_mobile_v2.0_SLANet English table recognition model trained on PubTabNet dataset based on SLANet 9M inference model / trained model
ch_ppstructure_mobile_v2.0_SLANet Chinese table recognition model trained on PubTabNet dataset based on SLANet 9.3M inference model / trained model

3. VQA

model description inference model size download
ser_LayoutXLM_xfun_zh SER model trained on xfun Chinese dataset based on LayoutXLM 1.4G inference model / trained model
re_LayoutXLM_xfun_zh Re model trained on xfun Chinese dataset based on LayoutXLM 1.4G inference model coming soon / trained model
ser_LayoutLMv2_xfun_zh SER model trained on xfun Chinese dataset based on LayoutXLMv2 778M inference model / trained model
re_LayoutLMv2_xfun_zh Re model trained on xfun Chinese dataset based on LayoutXLMv2 765M inference model coming soon / trained model
ser_LayoutLM_xfun_zh SER model trained on xfun Chinese dataset based on LayoutLM 430M inference model / trained model

4. KIE

model description model size download
SDMGR Key Information Extraction Model 78M inference model coming soon / trained model