api_ch.md 2.0 KB
Newer Older
W
WenmuZhou 已提交
1 2
# PaddleStructure

W
WenmuZhou 已提交
3
## 1. pipeline介绍
W
WenmuZhou 已提交
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18

PaddleStructure 是一个用于复杂板式文字OCR的工具包,流程如下
![pipeline](../doc/table/pipeline.png)

在PaddleStructure中,图片会先经由layoutparser进行版面分析,在版面分析中,会对图片里的区域进行分类,根据根据类别进行对于的ocr流程。

目前layoutparser会输出五个类别:
1. Text
2. Title
3. Figure
4. List
5. Table
   
1-4类走传统的OCR流程,5走表格的OCR流程。

W
WenmuZhou 已提交
19
## 2. LayoutParser
W
WenmuZhou 已提交
20

W
WenmuZhou 已提交
21
[文档](layout/README.md)
W
WenmuZhou 已提交
22

W
WenmuZhou 已提交
23
## 3. Table OCR
W
WenmuZhou 已提交
24 25 26

[文档](table/README_ch.md)

W
WenmuZhou 已提交
27
## 4. PaddleStructure whl包介绍
W
WenmuZhou 已提交
28

W
WenmuZhou 已提交
29
### 4.1 使用
W
WenmuZhou 已提交
30

W
WenmuZhou 已提交
31
4.1.1 代码使用
W
WenmuZhou 已提交
32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
```python
import cv2
from paddlestructure import PaddleStructure,draw_result

table_engine = PaddleStructure(
    output='./output/table',
    show_log=True)

img_path = '../doc/table/1.png'
img = cv2.imread(img_path)
result = table_engine(img)
for line in result:
    print(line)

from PIL import Image

font_path = 'path/tp/PaddleOCR/doc/fonts/simfang.ttf'
image = Image.open(img_path).convert('RGB')
im_show = draw_result(image, result,font_path=font_path)
im_show = Image.fromarray(im_show)
im_show.save('result.jpg')
```

W
WenmuZhou 已提交
55
4.1.2 命令行使用
W
WenmuZhou 已提交
56 57 58 59
```bash
paddlestructure --image_dir=../doc/table/1.png
```

W
WenmuZhou 已提交
60 61 62 63 64 65 66 67 68 69 70
### 参数说明
大部分参数和paddleocr whl包保持一致,见 [whl包文档](../doc/doc_ch/whl.md)

| 字段                    | 说明                                            | 默认值           |
|------------------------|------------------------------------------------------|------------------|
| output                 | excel和识别结果保存的地址                    | ./output/table            |
| structure_max_len      |  structure模型预测时,图像的长边resize尺度             |  488            |
| structure_model_dir      |  structure inference 模型地址             |  None            |
| structure_char_type      |  structure 模型所用字典地址             |  ../ppocr/utils/dict/table_structure_dict.tx            |