2022-04-25 16:03:14 +08:00
# PP-Structure Quick Start
2022-08-10 22:15:52 +08:00
- [1. Install package ](#1-install-package )
- [2. Use ](#2-use )
- [2.1 Use by command line ](#21-use-by-command-line )
2022-08-11 18:56:19 +08:00
- [2.1.1 image orientation + layout analysis + table recognition ](#211-image-orientation--layout-analysis--table-recognition )
- [2.1.2 layout analysis + table recognition ](#212-layout-analysis--table-recognition )
- [2.1.3 layout analysis ](#213-layout-analysis )
- [2.1.4 table recognition ](#214-table-recognition )
- [2.1.5 DocVQA ](#215-docvqa )
2022-08-10 22:15:52 +08:00
- [2.2 Use by code ](#22-use-by-code )
2022-08-11 18:56:19 +08:00
- [2.2.1 image orientation + layout analysis + table recognition ](#221-image-orientation--layout-analysis--table-recognition )
- [2.2.2 layout analysis + table recognition ](#222-layout-analysis--table-recognition )
- [2.2.3 layout analysis ](#223-layout-analysis )
- [2.2.4 table recognition ](#224-table-recognition )
- [2.2.5 DocVQA ](#225-docvqa )
2022-08-10 22:15:52 +08:00
- [2.3 Result description ](#23-result-description )
- [2.3.1 layout analysis + table recognition ](#231-layout-analysis--table-recognition )
- [2.3.2 DocVQA ](#232-docvqa )
- [2.4 Parameter Description ](#24-parameter-description )
2022-04-21 17:49:14 +08:00
< a name = "1" > < / a >
2022-04-25 16:03:14 +08:00
## 1. Install package
2022-04-21 17:49:14 +08:00
```bash
2022-04-25 16:03:14 +08:00
# Install paddleocr, version 2.5+ is recommended
2022-04-22 14:12:04 +08:00
pip3 install "paddleocr>=2.5"
2022-04-25 16:03:14 +08:00
# Install the DocVQA dependency package paddlenlp (if you do not use the DocVQA, you can skip it)
2022-04-21 17:49:14 +08:00
pip install paddlenlp
```
< a name = "2" > < / a >
2022-04-25 16:03:14 +08:00
## 2. Use
2022-04-21 17:49:14 +08:00
< a name = "21" > < / a >
2022-04-25 16:03:14 +08:00
### 2.1 Use by command line
2022-04-22 13:24:45 +08:00
2022-04-21 17:49:14 +08:00
< a name = "211" > < / a >
2022-08-11 18:56:19 +08:00
#### 2.1.1 image orientation + layout analysis + table recognition
2022-04-21 17:49:14 +08:00
```bash
2022-08-11 18:56:19 +08:00
paddleocr --image_dir=PaddleOCR/ppstructure/docs/table/1.png --type=structure --image_orientation=true
2022-04-21 17:49:14 +08:00
```
< a name = "212" > < / a >
2022-08-11 18:56:19 +08:00
#### 2.1.2 layout analysis + table recognition
2022-04-22 13:24:45 +08:00
```bash
2022-08-11 18:56:19 +08:00
paddleocr --image_dir=PaddleOCR/ppstructure/docs/table/1.png --type=structure
2022-04-22 13:24:45 +08:00
```
< a name = "213" > < / a >
2022-08-11 18:56:19 +08:00
#### 2.1.3 layout analysis
2022-04-22 13:24:45 +08:00
```bash
2022-08-11 18:56:19 +08:00
paddleocr --image_dir=PaddleOCR/ppstructure/docs/table/1.png --type=structure --table=false --ocr=false
2022-04-22 13:24:45 +08:00
```
< a name = "214" > < / a >
2022-08-11 18:56:19 +08:00
#### 2.1.4 table recognition
```bash
paddleocr --image_dir=PaddleOCR/ppstructure/docs/table/table.jpg --type=structure --layout=false
```
< a name = "215" > < / a >
#### 2.1.5 DocVQA
2022-04-21 17:49:14 +08:00
2022-04-25 16:03:14 +08:00
Please refer to: [Documentation Visual Q&A ](../vqa/README.md ) .
2022-04-21 17:49:14 +08:00
< a name = "22" > < / a >
2022-04-25 16:03:14 +08:00
### 2.2 Use by code
2022-04-21 17:49:14 +08:00
< a name = "221" > < / a >
2022-08-11 18:56:19 +08:00
#### 2.2.1 image orientation + layout analysis + table recognition
2022-04-21 17:49:14 +08:00
```python
import os
import cv2
from paddleocr import PPStructure,draw_structure_result,save_structure_res
2022-08-11 18:56:19 +08:00
table_engine = PPStructure(show_log=True, image_orientation=True)
2022-04-21 17:49:14 +08:00
2022-04-22 13:24:45 +08:00
save_folder = './output'
img_path = 'PaddleOCR/ppstructure/docs/table/1.png'
2022-04-21 17:49:14 +08:00
img = cv2.imread(img_path)
result = table_engine(img)
save_structure_res(result, save_folder,os.path.basename(img_path).split('.')[0])
for line in result:
line.pop('img')
print(line)
from PIL import Image
2022-04-22 13:24:45 +08:00
font_path = 'PaddleOCR/doc/fonts/simfang.ttf' # PaddleOCR下提供字体包
2022-04-21 17:49:14 +08:00
image = Image.open(img_path).convert('RGB')
im_show = draw_structure_result(image, result,font_path=font_path)
im_show = Image.fromarray(im_show)
im_show.save('result.jpg')
```
< a name = "222" > < / a >
2022-08-11 18:56:19 +08:00
#### 2.2.2 layout analysis + table recognition
```python
import os
import cv2
from paddleocr import PPStructure,draw_structure_result,save_structure_res
table_engine = PPStructure(show_log=True)
save_folder = './output'
img_path = 'PaddleOCR/ppstructure/docs/table/1.png'
img = cv2.imread(img_path)
result = table_engine(img)
save_structure_res(result, save_folder,os.path.basename(img_path).split('.')[0])
for line in result:
line.pop('img')
print(line)
from PIL import Image
font_path = 'PaddleOCR/doc/fonts/simfang.ttf' # PaddleOCR下提供字体包
image = Image.open(img_path).convert('RGB')
im_show = draw_structure_result(image, result,font_path=font_path)
im_show = Image.fromarray(im_show)
im_show.save('result.jpg')
```
< a name = "223" > < / a >
#### 2.2.3 layout analysis
2022-04-22 13:24:45 +08:00
```python
import os
import cv2
from paddleocr import PPStructure,save_structure_res
table_engine = PPStructure(table=False, ocr=False, show_log=True)
save_folder = './output'
img_path = 'PaddleOCR/ppstructure/docs/table/1.png'
img = cv2.imread(img_path)
result = table_engine(img)
save_structure_res(result, save_folder, os.path.basename(img_path).split('.')[0])
for line in result:
line.pop('img')
print(line)
```
2022-08-11 18:56:19 +08:00
< a name = "224" > < / a >
#### 2.2.4 table recognition
2022-04-22 13:24:45 +08:00
```python
import os
import cv2
from paddleocr import PPStructure,save_structure_res
table_engine = PPStructure(layout=False, show_log=True)
save_folder = './output'
img_path = 'PaddleOCR/ppstructure/docs/table/table.jpg'
img = cv2.imread(img_path)
result = table_engine(img)
save_structure_res(result, save_folder, os.path.basename(img_path).split('.')[0])
for line in result:
line.pop('img')
print(line)
```
2022-08-11 18:56:19 +08:00
< a name = "225" > < / a >
#### 2.2.5 DocVQA
2022-04-21 17:49:14 +08:00
2022-04-25 16:03:14 +08:00
Please refer to: [Documentation Visual Q&A ](../vqa/README.md ) .
2022-04-21 17:49:14 +08:00
< a name = "23" > < / a >
2022-04-25 16:03:14 +08:00
### 2.3 Result description
The return of PP-Structure is a list of dicts, the example is as follows:
2022-04-21 17:49:14 +08:00
< a name = "231" > < / a >
2022-04-25 16:03:14 +08:00
#### 2.3.1 layout analysis + table recognition
2022-04-21 17:49:14 +08:00
```shell
[
{ 'type': 'Text',
'bbox': [34, 432, 345, 462],
'res': ([[36.0, 437.0, 341.0, 437.0, 341.0, 446.0, 36.0, 447.0], [41.0, 454.0, 125.0, 453.0, 125.0, 459.0, 41.0, 460.0]],
[('Tigure-6. The performance of CNN and IPT models using difforen', 0.90060663), ('Tent ', 0.465441)])
}
]
```
2022-04-25 16:03:14 +08:00
Each field in dict is described as follows:
2022-04-21 17:49:14 +08:00
2022-08-11 18:56:19 +08:00
| field | description |
| --- |---|
2022-04-25 16:03:14 +08:00
|type| Type of image area. |
|bbox| The coordinates of the image area in the original image, respectively [upper left corner x, upper left corner y, lower right corner x, lower right corner y]. |
|res| OCR or table recognition result of the image area. < br > table: a dict with field descriptions as follows: < br >         `html` : html str of table.< br >         In the code usage mode, set return_ocr_result_in_table=True whrn call can get the detection and recognition results of each text in the table area, corresponding to the following fields: < br >         `boxes` : text detection boxes.< br >         `rec_res` : text recognition results.< br > OCR: A tuple containing the detection boxes and recognition results of each single text. |
2022-04-21 17:49:14 +08:00
2022-04-25 16:03:14 +08:00
After the recognition is completed, each image will have a directory with the same name under the directory specified by the `output` field. Each table in the image will be stored as an excel, and the picture area will be cropped and saved. The filename of excel and picture is their coordinates in the image.
2022-04-21 17:49:14 +08:00
```
/output/table/1/
└─ res.txt
2022-04-25 16:03:14 +08:00
└─ [454, 360, 824, 658].xlsx table recognition result
└─ [16, 2, 828, 305].jpg picture in Image
└─ [17, 361, 404, 711].xlsx table recognition result
2022-04-21 17:49:14 +08:00
```
< a name = "232" > < / a >
#### 2.3.2 DocVQA
2022-04-25 16:03:14 +08:00
Please refer to: [Documentation Visual Q&A ](../vqa/README.md ) .
2022-04-21 17:49:14 +08:00
< a name = "24" > < / a >
2022-04-25 16:03:14 +08:00
### 2.4 Parameter Description
2022-08-11 18:56:19 +08:00
| field | description | default |
|---|---|---|
| output | result save path | ./output/table |
| table_max_len | long side of the image resize in table structure model | 488 |
| table_model_dir | Table structure model inference model path| None |
| table_char_dict_path | The dictionary path of table structure model | ../ppocr/utils/dict/table_structure_dict.txt |
| merge_no_span_structure | In the table recognition model, whether to merge '\<td>' and '\</td>' | False |
| layout_model_dir | Layout analysis model inference model path| None |
| layout_dict_path | The dictionary path of layout analysis model| ../ppocr/utils/dict/layout_publaynet_dict.txt |
| layout_score_threshold | The box threshold path of layout analysis model| 0.5|
| layout_nms_threshold | The nms threshold path of layout analysis model| 0.5|
| vqa_algorithm | vqa model algorithm| LayoutXLM|
| ser_model_dir | Ser model inference model path| None|
| ser_dict_path | The dictionary path of Ser model| ../train_data/XFUND/class_list_xfun.txt|
| mode | structure or vqa | structure |
| image_orientation | Whether to perform image orientation classification in forward | False |
| layout | Whether to perform layout analysis in forward | True |
| table | Whether to perform table recognition in forward | True |
| ocr | Whether to perform ocr for non-table areas in layout analysis. When layout is False, it will be automatically set to False| True |
| recovery | Whether to perform layout recovery in forward| False |
| structure_version | Structure version, optional PP-structure and PP-structurev2 | PP-structure |
2022-04-25 16:03:14 +08:00
Most of the parameters are consistent with the PaddleOCR whl package, see [whl package documentation ](../../doc/doc_en/whl.md )