quickstart.md 9.2 KB
Newer Older
文幕地方's avatar
文幕地方 已提交
1 2
# PP-Structure 快速开始

文幕地方's avatar
文幕地方 已提交
3 4 5
- [1. 安装依赖包](#1-安装依赖包)
- [2. 便捷使用](#2-便捷使用)
  - [2.1 命令行使用](#21-命令行使用)
6
    - [2.1.1 图像方向分类+版面分析+表格识别](#211-图像方向分类版面分析表格识别)
文幕地方's avatar
文幕地方 已提交
7
    - [2.1.2 版面分析+表格识别](#212-版面分析表格识别)
8 9
    - [2.1.3 版面分析](#213-版面分析)
    - [2.1.4 表格识别](#214-表格识别)
littletomatodonkey's avatar
littletomatodonkey 已提交
10
    - [2.1.5 关键信息抽取](#215-关键信息抽取)
A
an1018 已提交
11
    - [2.1.6 版面恢复](#216-版面恢复)
文幕地方's avatar
文幕地方 已提交
12
  - [2.2 代码使用](#22-代码使用)
A
an1018 已提交
13 14

    - [2.2.1 图像方向+分类版面分析+表格识别](#221-图像方向分类版面分析表格识别)
15 16 17
    - [2.2.2 版面分析+表格识别](#222-版面分析表格识别)
    - [2.2.3 版面分析](#223-版面分析)
    - [2.2.4 表格识别](#224-表格识别)
A
an1018 已提交
18

littletomatodonkey's avatar
littletomatodonkey 已提交
19
    - [2.2.5 关键信息抽取](#225-关键信息抽取)
A
an1018 已提交
20
    - [2.2.6 版面恢复](#226-版面恢复)
A
an1018 已提交
21

文幕地方's avatar
文幕地方 已提交
22 23
  - [2.3 返回结果说明](#23-返回结果说明)
    - [2.3.1 版面分析+表格识别](#231-版面分析表格识别)
littletomatodonkey's avatar
littletomatodonkey 已提交
24
    - [2.3.2 关键信息抽取](#232-关键信息抽取)
A
an1018 已提交
25

文幕地方's avatar
文幕地方 已提交
26
  - [2.4 参数说明](#24-参数说明)
M
update  
MissPenguin 已提交
27 28 29


<a name="1"></a>
文幕地方's avatar
文幕地方 已提交
30 31 32
## 1. 安装依赖包

```bash
A
an1018 已提交
33 34
# 安装 paddleocr,推荐使用2.6版本
pip3 install "paddleocr>=2.6"
littletomatodonkey's avatar
littletomatodonkey 已提交
35 36
# 安装 关键信息抽取 依赖包(如不需要KIE功能,可跳过)
pip install -r kie/requirements.txt
A
an1018 已提交
37 38
# 安装 图像方向分类依赖包paddleclas(如不需要图像方向分类功能,可跳过)
pip3 install paddleclas
文幕地方's avatar
文幕地方 已提交
39 40
```

M
update  
MissPenguin 已提交
41
<a name="2"></a>
文幕地方's avatar
文幕地方 已提交
42 43
## 2. 便捷使用

M
update  
MissPenguin 已提交
44 45
<a name="21"></a>
### 2.1 命令行使用  
46

M
update  
MissPenguin 已提交
47
<a name="211"></a>
48 49 50 51 52 53
#### 2.1.1 图像方向分类+版面分析+表格识别
```bash
paddleocr --image_dir=PaddleOCR/ppstructure/docs/table/1.png --type=structure --image_orientation=true
```

<a name="212"></a>
文幕地方's avatar
文幕地方 已提交
54
#### 2.1.2 版面分析+表格识别
文幕地方's avatar
文幕地方 已提交
55
```bash
56
paddleocr --image_dir=PaddleOCR/ppstructure/docs/table/1.png --type=structure
文幕地方's avatar
文幕地方 已提交
57 58
```

59 60
<a name="213"></a>
#### 2.1.3 版面分析
61 62 63 64
```bash
paddleocr --image_dir=PaddleOCR/ppstructure/docs/table/1.png --type=structure --table=false --ocr=false
```

65 66
<a name="214"></a>
#### 2.1.4 表格识别
67 68 69 70
```bash
paddleocr --image_dir=PaddleOCR/ppstructure/docs/table/table.jpg --type=structure --layout=false
```

71
<a name="215"></a>
A
an1018 已提交
72

A
an1018 已提交
73
#### 2.1.5 关键信息抽取
littletomatodonkey's avatar
littletomatodonkey 已提交
74
请参考:[关键信息抽取教程](../kie/README_ch.md)
文幕地方's avatar
文幕地方 已提交
75

A
an1018 已提交
76 77 78 79 80 81 82 83
<a name="216"></a>

#### 2.1.6 版面恢复

```bash
paddleocr --image_dir=PaddleOCR/ppstructure/docs/table/1.png --type=structure --recovery=true
```

M
update  
MissPenguin 已提交
84
<a name="22"></a>
A
an1018 已提交
85

86
### 2.2 代码使用
文幕地方's avatar
文幕地方 已提交
87

M
update  
MissPenguin 已提交
88
<a name="221"></a>
A
an1018 已提交
89
#### 2.2.1 图像方向分类+版面分析+表格识别
M
update  
MissPenguin 已提交
90

文幕地方's avatar
文幕地方 已提交
91 92 93 94 95
```python
import os
import cv2
from paddleocr import PPStructure,draw_structure_result,save_structure_res

96
table_engine = PPStructure(show_log=True, image_orientation=True)
文幕地方's avatar
文幕地方 已提交
97

98 99
save_folder = './output'
img_path = 'PaddleOCR/ppstructure/docs/table/1.png'
文幕地方's avatar
文幕地方 已提交
100 101 102 103 104 105 106 107 108 109
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

110
font_path = 'PaddleOCR/doc/fonts/simfang.ttf' # PaddleOCR下提供字体包
文幕地方's avatar
文幕地方 已提交
111 112 113 114 115 116
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')
```

M
update  
MissPenguin 已提交
117
<a name="222"></a>
118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147
#### 2.2.2 版面分析+表格识别

```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 版面分析
148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166

```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)
```

167
<a name="224"></a>
A
an1018 已提交
168

169
#### 2.2.4 表格识别
170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188

```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)
```

189
<a name="225"></a>
littletomatodonkey's avatar
littletomatodonkey 已提交
190
#### 2.2.5 关键信息抽取
文幕地方's avatar
文幕地方 已提交
191

littletomatodonkey's avatar
littletomatodonkey 已提交
192
请参考:[关键信息抽取教程](../kie/README_ch.md)
文幕地方's avatar
文幕地方 已提交
193

A
an1018 已提交
194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220
<a name="226"></a>

#### 2.2.6 版面恢复

```python
import os
import cv2
from paddleocr import PPStructure,save_structure_res
from paddelocr.ppstructure.recovery.recovery_to_doc import sorted_layout_boxes, convert_info_docx

table_engine = PPStructure(layout=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)

h, w, _ = img.shape
res = sorted_layout_boxes(res, w)
convert_info_docx(img, result, save_folder, os.path.basename(img_path).split('.')[0])
```

M
update  
MissPenguin 已提交
221
<a name="23"></a>
文幕地方's avatar
文幕地方 已提交
222 223 224
### 2.3 返回结果说明
PP-Structure的返回结果为一个dict组成的list,示例如下

M
update  
MissPenguin 已提交
225 226
<a name="231"></a>
#### 2.3.1 版面分析+表格识别
文幕地方's avatar
文幕地方 已提交
227 228 229 230 231 232 233 234 235 236 237
```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)])
  }
]
```
dict 里各个字段说明如下

238 239 240 241
| 字段   | 说明|
| --- |---|
|type| 图片区域的类型 |
|bbox| 图片区域的在原图的坐标,分别[左上角x,左上角y,右下角x,右下角y]|
242
|res| 图片区域的OCR或表格识别结果。<br> 表格: 一个dict,字段说明如下<br>&emsp;&emsp;&emsp;&emsp;&emsp;&emsp;&emsp; `html`: 表格的HTML字符串<br>&emsp;&emsp;&emsp;&emsp;&emsp;&emsp;&emsp; 在代码使用模式下,前向传入return_ocr_result_in_table=True可以拿到表格中每个文本的检测识别结果,对应为如下字段: <br>&emsp;&emsp;&emsp;&emsp;&emsp;&emsp;&emsp; `boxes`: 文本检测坐标<br>&emsp;&emsp;&emsp;&emsp;&emsp;&emsp;&emsp; `rec_res`: 文本识别结果。<br> OCR: 一个包含各个单行文字的检测坐标和识别结果的元组 |
文幕地方's avatar
文幕地方 已提交
243

M
update  
MissPenguin 已提交
244 245 246 247 248 249
运行完成后,每张图片会在`output`字段指定的目录下有一个同名目录,图片里的每个表格会存储为一个excel,图片区域会被裁剪之后保存下来,excel文件和图片名为表格在图片里的坐标。

  ```
  /output/table/1/
    └─ res.txt
    └─ [454, 360, 824, 658].xlsx  表格识别结果
250 251
    └─ [16, 2, 828, 305].jpg            被裁剪出的图片区域
    └─ [17, 361, 404, 711].xlsx        表格识别结果
M
update  
MissPenguin 已提交
252 253 254
  ```

<a name="232"></a>
littletomatodonkey's avatar
littletomatodonkey 已提交
255
#### 2.3.2 关键信息抽取
文幕地方's avatar
文幕地方 已提交
256

littletomatodonkey's avatar
littletomatodonkey 已提交
257
请参考:[关键信息抽取教程](../kie/README_ch.md)
文幕地方's avatar
文幕地方 已提交
258

M
update  
MissPenguin 已提交
259
<a name="24"></a>
文幕地方's avatar
文幕地方 已提交
260 261
### 2.4 参数说明

262 263 264 265 266 267 268 269 270 271 272
| 字段 | 说明  | 默认值  |
|---|---|---|
| output | 结果保存地址 | ./output/table |
| table_max_len | 表格结构模型预测时,图像的长边resize尺度 | 488 |
| table_model_dir | 表格结构模型 inference 模型地址| None |
| table_char_dict_path | 表格结构模型所用字典地址 | ../ppocr/utils/dict/table_structure_dict.txt  |
| merge_no_span_structure | 表格识别模型中,是否对'\<td>'和'\</td>' 进行合并 | False |
| layout_model_dir  | 版面分析模型 inference 模型地址 | None |
| layout_dict_path  | 版面分析模型字典| ../ppocr/utils/dict/layout_publaynet_dict.txt |
| layout_score_threshold  | 版面分析模型检测框阈值| 0.5|
| layout_nms_threshold  | 版面分析模型nms阈值| 0.5|
273
| kie_algorithm  | kie模型算法| LayoutXLM|
274 275
| ser_model_dir  | ser模型  inference 模型地址| None|
| ser_dict_path  | ser模型字典| ../train_data/XFUND/class_list_xfun.txt|
276
| mode | structure or kie  | structure   |
277 278 279 280 281
| image_orientation | 前向中是否执行图像方向分类  | False   |
| layout | 前向中是否执行版面分析  | True   |
| table  | 前向中是否执行表格识别  | True   |
| ocr    | 对于版面分析中的非表格区域,是否执行ocr。当layout为False时会被自动设置为False| True |
| recovery    | 前向中是否执行版面恢复| False |
A
an1018 已提交
282
| save_pdf | 版面恢复导出docx文件的同时,是否导出pdf文件 | False |
283
| structure_version |  模型版本,可选 PP-structure和PP-structurev2  | PP-structure |
文幕地方's avatar
文幕地方 已提交
284

fanruinet's avatar
fanruinet 已提交
285
大部分参数和PaddleOCR whl包保持一致,见 [whl包文档](../../doc/doc_ch/whl.md)