README.md 6.5 KB
Newer Older
G
grasswolfs 已提交
1
English | [简体中文](README_ch.md)
W
WenmuZhou 已提交
2

文幕地方's avatar
update  
文幕地方 已提交
3 4 5 6 7 8 9 10 11 12 13
- [1. Introduction](#1)
- [2. Update log](#2)
- [3. Features](#3)
- [4. Results](#4)
  * [4.1 Layout analysis and table recognition](#41)
  * [4.2 DOC-VQA](#42)
- [5. Quick start](#5)
- [6. PP-Structure System](#6)
  * [6.1 Layout analysis and table recognition](#61)
  * [6.2 DOC-VQA](#62)
- [7. Model List](#7)
W
opt doc  
WenmuZhou 已提交
14

文幕地方's avatar
update  
文幕地方 已提交
15
<a name="1"></a>
16

文幕地方's avatar
update  
文幕地方 已提交
17
## 1. Introduction
18

文幕地方's avatar
update  
文幕地方 已提交
19
PP-Structure is an OCR toolkit that can be used for document analysis and processing with complex structures, designed to help developers better complete document understanding tasks
20

文幕地方's avatar
update  
文幕地方 已提交
21
<a name="2"></a>
22

文幕地方's avatar
update  
文幕地方 已提交
23 24
## 2. Update log
* 2021.12.07 add [DOC-VQA SER and RE tasks](vqa/README.md)
G
grasswolfs 已提交
25

文幕地方's avatar
update  
文幕地方 已提交
26
<a name="3"></a>
27

文幕地方's avatar
update  
文幕地方 已提交
28
## 3. Features
29

文幕地方's avatar
update  
文幕地方 已提交
30
The main features of PP-Structure are as follows:
G
grasswolfs 已提交
31

文幕地方's avatar
update  
文幕地方 已提交
32 33 34 35 36 37
- Support the layout analysis of documents, divide the documents into 5 types of areas **text, title, table, image and list** (conjunction with Layout-Parser)
- Support to extract the texts from the text, title, picture and list areas (used in conjunction with PP-OCR)
- Support to extract excel files from the table areas
- Support python whl package and command line usage, easy to use
- Support custom training for layout analysis and table structure tasks
- Support Document Visual Question Answering (DOC-VQA) tasks: Semantic Entity Recognition (SER) and Relation Extraction (RE)
38

W
opt doc  
WenmuZhou 已提交
39

文幕地方's avatar
update  
文幕地方 已提交
40
<a name="4"></a>
文幕地方's avatar
文幕地方 已提交
41

文幕地方's avatar
update  
文幕地方 已提交
42
## 4. Results
43

文幕地方's avatar
update  
文幕地方 已提交
44
<a name="41"></a>
G
grasswolfs 已提交
45

文幕地方's avatar
update  
文幕地方 已提交
46
### 4.1 Layout analysis and table recognition
W
WenmuZhou 已提交
47

文幕地方's avatar
update  
文幕地方 已提交
48
<img src="../doc/table/ppstructure.GIF" width="100%"/>
G
grasswolfs 已提交
49

文幕地方's avatar
update  
文幕地方 已提交
50
The figure shows the pipeline of layout analysis + table recognition. The image is first divided into four areas of image, text, title and table by layout analysis, and then OCR detection and recognition is performed on the three areas of image, text and title, and the table is performed table recognition, where the image will also be stored for use.
G
grasswolfs 已提交
51

文幕地方's avatar
update  
文幕地方 已提交
52
<a name="42"></a>
W
WenmuZhou 已提交
53

文幕地方's avatar
update  
文幕地方 已提交
54
### 4.2 DOC-VQA
W
WenmuZhou 已提交
55

文幕地方's avatar
update  
文幕地方 已提交
56
* SER
G
grasswolfs 已提交
57

文幕地方's avatar
update  
文幕地方 已提交
58 59
![](./vqa/images/result_ser/zh_val_0_ser.jpg) | ![](./vqa/images/result_ser/zh_val_42_ser.jpg)
---|---
W
WenmuZhou 已提交
60

文幕地方's avatar
update  
文幕地方 已提交
61
Different colored boxes in the figure represent different categories. For xfun dataset, there are three categories: query, answer and header:
W
opt doc  
WenmuZhou 已提交
62

文幕地方's avatar
update  
文幕地方 已提交
63 64 65
* Dark purple: header
* Light purple: query
* Army green: answer
W
WenmuZhou 已提交
66

文幕地方's avatar
update  
文幕地方 已提交
67
The corresponding category and OCR recognition results are also marked at the top left of the OCR detection box.
W
WenmuZhou 已提交
68 69


文幕地方's avatar
update  
文幕地方 已提交
70
* RE
W
WenmuZhou 已提交
71

文幕地方's avatar
update  
文幕地方 已提交
72 73
![](./vqa/images/result_re/zh_val_21_re.jpg) | ![](./vqa/images/result_re/zh_val_40_re.jpg)
---|---
W
WenmuZhou 已提交
74 75


文幕地方's avatar
update  
文幕地方 已提交
76
In the figure, the red box represents the question, the blue box represents the answer, and the question and answer are connected by green lines. The corresponding category and OCR recognition results are also marked at the top left of the OCR detection box.
W
WenmuZhou 已提交
77 78


文幕地方's avatar
update  
文幕地方 已提交
79
<a name="5"></a>
W
WenmuZhou 已提交
80

文幕地方's avatar
update  
文幕地方 已提交
81
## 5. Quick start
W
WenmuZhou 已提交
82

文幕地方's avatar
update  
文幕地方 已提交
83
Start from [Quick Installation](./docs/quickstart.md)
W
opt doc  
WenmuZhou 已提交
84

文幕地方's avatar
update  
文幕地方 已提交
85
<a name="6"></a>
W
opt doc  
WenmuZhou 已提交
86

文幕地方's avatar
update  
文幕地方 已提交
87
## 6. PP-Structure System
W
opt doc  
WenmuZhou 已提交
88

文幕地方's avatar
update  
文幕地方 已提交
89
<a name="61"></a>
W
opt doc  
WenmuZhou 已提交
90

文幕地方's avatar
update  
文幕地方 已提交
91
### 6.1 Layout analysis and table recognition
W
opt doc  
WenmuZhou 已提交
92

文幕地方's avatar
update  
文幕地方 已提交
93
![pipeline](../doc/table/pipeline.jpg)
W
WenmuZhou 已提交
94

文幕地方's avatar
update  
文幕地方 已提交
95
In PP-Structure, the image will be divided into 5 types of areas **text, title, image list and table**. For the first 4 types of areas, directly use PP-OCR system to complete the text detection and recognition. For the table area, after the table structuring process, the table in image is converted into an Excel file with the same table style.
W
opt doc  
WenmuZhou 已提交
96

文幕地方's avatar
update  
文幕地方 已提交
97
#### 6.1.1 Layout analysis
W
opt doc  
WenmuZhou 已提交
98

文幕地方's avatar
update  
文幕地方 已提交
99
Layout analysis classifies image by region, including the use of Python scripts of layout analysis tools, extraction of designated category detection boxes, performance indicators, and custom training layout analysis models. For details, please refer to [document](layout/README.md).
W
opt doc  
WenmuZhou 已提交
100

文幕地方's avatar
update  
文幕地方 已提交
101
#### 6.1.2 Table recognition
W
opt doc  
WenmuZhou 已提交
102

文幕地方's avatar
update  
文幕地方 已提交
103
Table recognition converts table images into excel documents, which include the detection and recognition of table text and the prediction of table structure and cell coordinates. For detailed instructions, please refer to [document](table/README.md)
W
opt doc  
WenmuZhou 已提交
104

文幕地方's avatar
update  
文幕地方 已提交
105
<a name="62"></a>
W
opt doc  
WenmuZhou 已提交
106

文幕地方's avatar
update  
文幕地方 已提交
107
### 6.2 DOC-VQA
W
WenmuZhou 已提交
108

文幕地方's avatar
update  
文幕地方 已提交
109
Document Visual Question Answering (DOC-VQA) if a type of Visual Question Answering (VQA), which includes Semantic Entity Recognition (SER) and Relation Extraction (RE) tasks. Based on SER task, text recognition and classification in images can be completed. Based on THE RE task, we can extract the relation of the text content in the image, such as judge the problem pair. For details, please refer to [document](vqa/README.md)
W
WenmuZhou 已提交
110

W
WenmuZhou 已提交
111

文幕地方's avatar
update  
文幕地方 已提交
112
<a name="7"></a>
W
WenmuZhou 已提交
113

文幕地方's avatar
update  
文幕地方 已提交
114
## 7. Model List
115

文幕地方's avatar
update  
文幕地方 已提交
116
PP-Structure系列模型列表(更新中)
117

文幕地方's avatar
update  
文幕地方 已提交
118
* Layout analysis model
119 120 121

|model name|description|download|
| --- | --- | --- |
文幕地方's avatar
update  
文幕地方 已提交
122
| ppyolov2_r50vd_dcn_365e_publaynet | The layout analysis model trained on the PubLayNet dataset can divide image into 5 types of areas **text, title, table, picture, and list** | [PubLayNet](https://paddle-model-ecology.bj.bcebos.com/model/layout-parser/ppyolov2_r50vd_dcn_365e_publaynet.tar) |
123

文幕地方's avatar
update  
文幕地方 已提交
124 125

* OCR and table recognition model
126 127 128

|model name|description|model size|download|
| --- | --- | --- | --- |
W
WenmuZhou 已提交
129 130
|ch_ppocr_mobile_slim_v2.0_det|Slim pruned lightweight model, supporting Chinese, English, multilingual text detection|2.6M|[inference model](https://paddleocr.bj.bcebos.com/dygraph_v2.0/slim/ch_ppocr_mobile_v2.0_det_prune_infer.tar) / [trained model](https://paddleocr.bj.bcebos.com/dygraph_v2.0/slim/ch_ppocr_mobile_v2.0_det_prune_infer.tar) |
|ch_ppocr_mobile_slim_v2.0_rec|Slim pruned and quantized lightweight model, supporting Chinese, English and number recognition|6M|[inference model](https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_slim_infer.tar) / [trained model](https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_slim_train.tar) |
文幕地方's avatar
update  
文幕地方 已提交
131 132 133
|en_ppocr_mobile_v2.0_table_structure|Table structure prediction of English table scene trained on PubLayNet dataset|[inference model](https://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_structure_infer.tar) / [trained model](https://paddleocr.bj.bcebos.com/dygraph_v2.1/table/en_ppocr_mobile_v2.0_table_structure_train.tar) |

* VQA model
134

文幕地方's avatar
update  
文幕地方 已提交
135 136 137 138
|model name|description|model size|download|
| --- | --- | --- | --- |
|PP-Layout_v1.0_ser_pretrained|SER model trained on xfun Chinese dataset based on LayoutXLM|1.4G|[inference model coming soon]() / [trained model](https://paddleocr.bj.bcebos.com/pplayout/PP-Layout_v1.0_ser_pretrained.tar) |
|PP-Layout_v1.0_re_pretrained|RE model trained on xfun Chinese dataset based on LayoutXLM|1.4G|[inference model coming soon]() / [trained model](https://paddleocr.bj.bcebos.com/pplayout/PP-Layout_v1.0_re_pretrained.tar) |
L
LDOUBLEV 已提交
139

文幕地方's avatar
update  
文幕地方 已提交
140
If you need to use other models, you can download the model in [PPOCR model_list](../doc/doc_en/models_list_en.md) and  [PPStructure model_list](./docs/model_list.md)