Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
PaddleOCR
提交
0a410708
P
PaddleOCR
项目概览
PaddlePaddle
/
PaddleOCR
大约 1 年 前同步成功
通知
1528
Star
32962
Fork
6643
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
108
列表
看板
标记
里程碑
合并请求
7
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
PaddleOCR
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
108
Issue
108
列表
看板
标记
里程碑
合并请求
7
合并请求
7
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
提交
0a410708
编写于
12月 13, 2021
作者:
文幕地方
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add sub doc
上级
e16ae81e
变更
4
显示空白变更内容
内联
并排
Showing
4 changed file
with
233 addition
and
190 deletion
+233
-190
ppstructure/README_ch.md
ppstructure/README_ch.md
+6
-190
ppstructure/docs/installation.md
ppstructure/docs/installation.md
+28
-0
ppstructure/docs/model_list.md
ppstructure/docs/model_list.md
+28
-0
ppstructure/docs/quickstart.md
ppstructure/docs/quickstart.md
+171
-0
未找到文件。
ppstructure/README_ch.md
浏览文件 @
0a410708
...
@@ -41,207 +41,23 @@ PP-Structure是一个可用于复杂文档结构分析和处理的OCR工具包
...
@@ -41,207 +41,23 @@ PP-Structure是一个可用于复杂文档结构分析和处理的OCR工具包
图中红色框表示问题,蓝色框表示答案,问题和答案之间使用绿色线连接。在OCR检测框的左上方也标出了对应的类别和OCR识别结果。
图中红色框表示问题,蓝色框表示答案,问题和答案之间使用绿色线连接。在OCR检测框的左上方也标出了对应的类别和OCR识别结果。
##
2. 安装
##
3. PP-Structure Pipeline介绍
### 2.1 安装依赖
### 3.1 版面分析+表格识别
-
**(1) 安装PaddlePaddle**
```
bash
pip3
install
--upgrade
pip
# GPU安装
python3
-m
pip
install
paddlepaddle-gpu
==
2.1.1
-i
https://mirror.baidu.com/pypi/simple
# CPU安装
python3
-m
pip
install
paddlepaddle
==
2.1.1
-i
https://mirror.baidu.com/pypi/simple
```
更多需求,请参照
[
安装文档
](
https://www.paddlepaddle.org.cn/install/quick
)
中的说明进行操作。
-
**(2) 安装依赖 **
```
bash
# 版面分析所需 Layout-Parser
pip3
install
-U
https://paddleocr.bj.bcebos.com/whl/layoutparser-0.0.0-py3-none-any.whl
# VQA所需 PaddleNLP
git clone https://github.com/PaddlePaddle/PaddleNLP
-b
develop
cd
PaddleNLP
pip3
install
-e
.
```
### 2.2 安装PaddleOCR(包含PP-OCR和PP-Structure)
-
**(1) PIP快速安装PaddleOCR whl包(仅预测)**
```
bash
pip3
install
"paddleocr>=2.2"
# 推荐使用2.2+版本
```
-
**(2) 完整克隆PaddleOCR源码(预测+训练)**
```
bash
【推荐】git clone https://github.com/PaddlePaddle/PaddleOCR
#如果因为网络问题无法pull成功,也可选择使用码云上的托管:
git clone https://gitee.com/paddlepaddle/PaddleOCR
#注:码云托管代码可能无法实时同步本github项目更新,存在3~5天延时,请优先使用推荐方式。
```
## 3. PP-Structure 快速开始
### 3.1 命令行使用(默认参数,极简)
*
版面分析+表格识别
```
bash
paddleocr
--image_dir
=
../doc/table/1.png
--type
=
structure
```
### 3.2 Python脚本使用(自定义参数,灵活)
*
版面分析+表格识别
```
python
import
os
import
cv2
from
paddleocr
import
PPStructure
,
draw_structure_result
,
save_structure_res
table_engine
=
PPStructure
(
show_log
=
True
)
save_folder
=
'./output/table'
img_path
=
'../doc/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
=
'../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'
)
```
### 3.3 返回结果说明
PP-Structure的返回结果为一个dict组成的list,示例如下
*
版面分析+表格识别
```
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 里各个字段说明如下
| 字段 | 说明 |
| --------------- | -------------|
|type|图片区域的类型|
|bbox|图片区域的在原图的坐标,分别[左上角x,左上角y,右下角x,右下角y]|
|res|图片区域的OCR或表格识别结果。
<br>
表格: 表格的HTML字符串;
<br>
OCR: 一个包含各个单行文字的检测坐标和识别结果的元组|
### 3.4 参数说明
| 字段 | 说明 | 默认值 |
| --------------- | ---------------------------------------- | ------------------------------------------- |
| output | excel和识别结果保存的地址 | ./output/table |
| table_max_len | 表格结构模型预测时,图像的长边resize尺度 | 488 |
| table_model_dir | 表格结构模型 inference 模型地址 | None |
| table_char_type | 表格结构模型所用字典地址 | ../ppocr/utils/dict/table_structure_dict.tx |
大部分参数和paddleocr whl包保持一致,见
[
whl包文档
](
../doc/doc_ch/whl.md
)
运行完成后,每张图片会在
`output`
字段指定的目录下有一个同名目录,图片里的每个表格会存储为一个excel,图片区域会被裁剪之后保存下来,excel文件和图片名名为表格在图片里的坐标。
## 4. PP-Structure 版面分析+表格识别 Pipeline介绍
![
pipeline
](
../doc/table/pipeline.jpg
)
![
pipeline
](
../doc/table/pipeline.jpg
)
在PP-Structure中,图片会先经由Layout-Parser进行版面分析,在版面分析中,会对图片里的区域进行分类,包括
**文字、标题、图片、列表和表格**
5类。对于前4类区域,直接使用PP-OCR完成对应区域文字检测与识别。对于表格类区域,经过表格结构化处理后,表格图片转换为相同表格样式的Excel文件。
在PP-Structure中,图片会先经由Layout-Parser进行版面分析,在版面分析中,会对图片里的区域进行分类,包括
**文字、标题、图片、列表和表格**
5类。对于前4类区域,直接使用PP-OCR完成对应区域文字检测与识别。对于表格类区域,经过表格结构化处理后,表格图片转换为相同表格样式的Excel文件。
###
4
.1 版面分析
###
# 3.1
.1 版面分析
版面分析对文档数据进行区域分类,其中包括版面分析工具的Python脚本使用、提取指定类别检测框、性能指标以及自定义训练版面分析模型,详细内容可以参考
[
文档
](
layout/README_ch.md
)
。
版面分析对文档数据进行区域分类,其中包括版面分析工具的Python脚本使用、提取指定类别检测框、性能指标以及自定义训练版面分析模型,详细内容可以参考
[
文档
](
layout/README_ch.md
)
。
###
4
.2 表格识别
###
# 3.1
.2 表格识别
表格识别将表格图片转换为excel文档,其中包含对于表格文本的检测和识别以及对于表格结构和单元格坐标的预测,详细说明参考
[
文档
](
table/README_ch.md
)
表格识别将表格图片转换为excel文档,其中包含对于表格文本的检测和识别以及对于表格结构和单元格坐标的预测,详细说明参考
[
文档
](
table/README_ch.md
)
## 5. 预测引擎推理(与whl包效果相同)
使用如下命令即可完成预测引擎的推理
*
版面分析+表格识别
```
python
cd
ppstructure
# 下载模型
mkdir
inference
&&
cd
inference
# 下载超轻量级中文OCR模型的检测模型并解压
wget
https
:
//
paddleocr
.
bj
.
bcebos
.
com
/
dygraph_v2
.
0
/
ch
/
ch_ppocr_mobile_v2
.
0
_det_infer
.
tar
&&
tar
xf
ch_ppocr_mobile_v2
.
0
_det_infer
.
tar
# 下载超轻量级中文OCR模型的识别模型并解压
wget
https
:
//
paddleocr
.
bj
.
bcebos
.
com
/
dygraph_v2
.
0
/
ch
/
ch_ppocr_mobile_v2
.
0
_rec_infer
.
tar
&&
tar
xf
ch_ppocr_mobile_v2
.
0
_rec_infer
.
tar
# 下载超轻量级英文表格英寸模型并解压
wget
https
:
//
paddleocr
.
bj
.
bcebos
.
com
/
dygraph_v2
.
0
/
table
/
en_ppocr_mobile_v2
.
0
_table_structure_infer
.
tar
&&
tar
xf
en_ppocr_mobile_v2
.
0
_table_structure_infer
.
tar
cd
..
python3
predict_system
.
py
--
det_model_dir
=
inference
/
ch_ppocr_mobile_v2
.
0
_det_infer
--
rec_model_dir
=
inference
/
ch_ppocr_mobile_v2
.
0
_rec_infer
--
table_model_dir
=
inference
/
en_ppocr_mobile_v2
.
0
_table_structure_infer
--
image_dir
=
..
/
doc
/
table
/
1.
png
--
rec_char_dict_path
=
..
/
ppocr
/
utils
/
ppocr_keys_v1
.
txt
--
table_char_dict_path
=
..
/
ppocr
/
utils
/
dict
/
table_structure_dict
.
txt
--
output
=
..
/
output
/
table
--
vis_font_path
=
..
/
doc
/
fonts
/
simfang
.
ttf
```
运行完成后,每张图片会在
`output`
字段指定的目录下的
`talbe`
目录下有一个同名目录,图片里的每个表格会存储为一个excel,图片区域会被裁剪之后保存下来,excel文件和图片名名为表格在图片里的坐标。
*
VQA
```
python
cd
ppstructure
# 下载模型
mkdir
inference
&&
cd
inference
# 下载SER xfun 模型并解压
wget
https
:
//
paddleocr
.
bj
.
bcebos
.
com
/
pplayout
/
PP
-
Layout_v1
.
0
_ser_pretrained
.
tar
&&
tar
xf
PP
-
Layout_v1
.
0
_ser_pretrained
.
tar
cd
..
python3
predict_system
.
py
--
model_name_or_path
=
vqa
/
PP
-
Layout_v1
.
0
_ser_pretrained
/
--
mode
=
vqa
--
image_dir
=
vqa
/
images
/
input
/
zh_val_0
.
jpg
--
vis_font_path
=
..
/
doc
/
fonts
/
simfang
.
ttf
```
运行完成后,每张图片会在
`output`
字段指定的目录下的
`vqa`
目录下存放可视化之后的图片,图片名和输入图片名一致。
**Model List**
LayoutParser 模型
|模型名称|模型简介|下载地址|
| --- | --- | --- |
| ppyolov2_r50vd_dcn_365e_publaynet | PubLayNet 数据集训练的版面分析模型,可以划分
**文字、标题、表格、图片以及列表**
5类区域 |
[
PubLayNet
](
https://paddle-model-ecology.bj.bcebos.com/model/layout-parser/ppyolov2_r50vd_dcn_365e_publaynet.tar
)
|
| ppyolov2_r50vd_dcn_365e_tableBank_word | TableBank Word 数据集训练的版面分析模型,只能检测表格 |
[
TableBank Word
](
https://paddle-model-ecology.bj.bcebos.com/model/layout-parser/ppyolov2_r50vd_dcn_365e_tableBank_word.tar
)
|
| ppyolov2_r50vd_dcn_365e_tableBank_latex | TableBank Latex 数据集训练的版面分析模型,只能检测表格 |
[
TableBank Latex
](
https://paddle-model-ecology.bj.bcebos.com/model/layout-parser/ppyolov2_r50vd_dcn_365e_tableBank_latex.tar
)
|
OCR和表格识别模型
|模型名称|模型简介|推理模型大小|下载地址|
| --- | --- | --- | --- |
|ch_ppocr_mobile_slim_v2.0_det|slim裁剪版超轻量模型,支持中英文、多语种文本检测|2.6M|
[
推理模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/slim/ch_ppocr_mobile_v2.0_det_prune_infer.tar
)
/
[
训练模型
](
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裁剪量化版超轻量模型,支持中英文、数字识别|6M|
[
推理模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_slim_infer.tar
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_slim_train.tar
)
|
|en_ppocr_mobile_v2.0_table_det|PubLayNet数据集训练的英文表格场景的文字检测|4.7M|
[
推理模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_det_infer.tar
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.1/table/en_ppocr_mobile_v2.0_table_det_train.tar
)
|
|en_ppocr_mobile_v2.0_table_rec|PubLayNet数据集训练的英文表格场景的文字识别|6.9M|
[
推理模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_rec_infer.tar
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.1/table/en_ppocr_mobile_v2.0_table_rec_train.tar
)
|
|en_ppocr_mobile_v2.0_table_structure|PubLayNet数据集训练的英文表格场景的表格结构预测|18.6M|
[
推理模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_structure_infer.tar
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.1/table/en_ppocr_mobile_v2.0_table_structure_train.tar
)
|
VQA
|模型名称|模型简介|推理模型大小|下载地址|
### 3.2 VQA
| --- | --- | --- | --- |
|PP-Layout_v1.0_ser_pretrained|基于LayoutXLM在xfun中文数据集上训练的SER模型|1.4G|
[
coming soon
](
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/pplayout/PP-Layout_v1.0_ser_pretrained.tar
)
|
|PP-Layout_v1.0_re_pretrained|基于LayoutXLM在xfun中文数据集上训练的RE模型|1.4G|
[
coming soon
](
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/pplayout/PP-Layout_v1.0_re_pretrained.tar
)
|
如需要使用其他模型,可以在
[
model_list
](
../doc/doc_ch/models_list.md
)
下载模型或者使用自己训练好的模型配置到
`det_model_dir`
,
`rec_model_dir`
,
`table_model_dir`
三个字段即可。
coming soon
ppstructure/docs/installation.md
0 → 100644
浏览文件 @
0a410708
# 快速安装
## 1. PaddlePaddle 和 PaddleOCR
可参考
[
PaddleOCR安装文档
](
../../doc/doc_ch/installation.md
)
## 2. 安装其他依赖
### 2.1 版面分析所需 Layout-Parser
Layout-Parser 可通过如下命令安装
```
bash
pip3
install
-U
https://paddleocr.bj.bcebos.com/whl/layoutparser-0.0.0-py3-none-any.whl
```
### 2.2 VQA所需依赖
*
paddleocr
```
bash
pip3
install
paddleocr
```
*
PaddleNLP
```
bash
git clone https://github.com/PaddlePaddle/PaddleNLP
-b
develop
cd
PaddleNLP
pip3
install
-e
.
```
ppstructure/docs/model_list.md
0 → 100644
浏览文件 @
0a410708
# Model List
## 1. LayoutParser 模型
|模型名称|模型简介|下载地址|
| --- | --- | --- |
| ppyolov2_r50vd_dcn_365e_publaynet | PubLayNet 数据集训练的版面分析模型,可以划分
**文字、标题、表格、图片以及列表**
5类区域 |
[
PubLayNet
](
https://paddle-model-ecology.bj.bcebos.com/model/layout-parser/ppyolov2_r50vd_dcn_365e_publaynet.tar
)
|
| ppyolov2_r50vd_dcn_365e_tableBank_word | TableBank Word 数据集训练的版面分析模型,只能检测表格 |
[
TableBank Word
](
https://paddle-model-ecology.bj.bcebos.com/model/layout-parser/ppyolov2_r50vd_dcn_365e_tableBank_word.tar
)
|
| ppyolov2_r50vd_dcn_365e_tableBank_latex | TableBank Latex 数据集训练的版面分析模型,只能检测表格 |
[
TableBank Latex
](
https://paddle-model-ecology.bj.bcebos.com/model/layout-parser/ppyolov2_r50vd_dcn_365e_tableBank_latex.tar
)
|
## 2. OCR和表格识别模型
|模型名称|模型简介|推理模型大小|下载地址|
| --- | --- | --- | --- |
|ch_ppocr_mobile_slim_v2.0_det|slim裁剪版超轻量模型,支持中英文、多语种文本检测|2.6M|
[
推理模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/slim/ch_ppocr_mobile_v2.0_det_prune_infer.tar
)
/
[
训练模型
](
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裁剪量化版超轻量模型,支持中英文、数字识别|6M|
[
推理模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_slim_infer.tar
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_slim_train.tar
)
|
|en_ppocr_mobile_v2.0_table_det|PubLayNet数据集训练的英文表格场景的文字检测|4.7M|
[
推理模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_det_infer.tar
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.1/table/en_ppocr_mobile_v2.0_table_det_train.tar
)
|
|en_ppocr_mobile_v2.0_table_rec|PubLayNet数据集训练的英文表格场景的文字识别|6.9M|
[
推理模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_rec_infer.tar
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.1/table/en_ppocr_mobile_v2.0_table_rec_train.tar
)
|
|en_ppocr_mobile_v2.0_table_structure|PubLayNet数据集训练的英文表格场景的表格结构预测|18.6M|
[
推理模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_structure_infer.tar
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/dygraph_v2.1/table/en_ppocr_mobile_v2.0_table_structure_train.tar
)
|
如需要使用其他OCR模型,可以在
[
model_list
](
../../doc/doc_ch/models_list.md
)
下载模型或者使用自己训练好的模型配置到
`det_model_dir`
,
`rec_model_dir`
两个字段即可。
## 3. VQA模型
|模型名称|模型简介|推理模型大小|下载地址|
| --- | --- | --- | --- |
|PP-Layout_v1.0_ser_pretrained|基于LayoutXLM在xfun中文数据集上训练的SER模型|1.4G|
[
推理模型 coming soon
](
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/pplayout/PP-Layout_v1.0_ser_pretrained.tar
)
|
|PP-Layout_v1.0_re_pretrained|基于LayoutXLM在xfun中文数据集上训练的RE模型|1.4G|
[
推理模型 coming soon
](
)
/
[
训练模型
](
https://paddleocr.bj.bcebos.com/pplayout/PP-Layout_v1.0_re_pretrained.tar
)
|
ppstructure/docs/quickstart.md
0 → 100644
浏览文件 @
0a410708
# PP-Structure 快速开始
*
[
1. 安装PaddleOCR whl包
](
#1
)
*
[
2. 便捷使用
](
#2
)
+
[
2.1 命令行使用
](
#21
)
+
[
2.2 Python脚本使用
](
#22
)
+
[
2.3 返回结果说明
](
#23
)
+
[
2.4 参数说明
](
#24
)
*
[
3. Python脚本使用
](
#3
)
<a
name=
"1"
></a>
## 1. 安装依赖包
```
bash
pip
install
"paddleocr>=2.3.0.2"
# 推荐使用2.3.0.2+版本
pip3
install
-U
https://paddleocr.bj.bcebos.com/whl/layoutparser-0.0.0-py3-none-any.whl
# 安装 PaddleNLP
git clone https://github.com/PaddlePaddle/PaddleNLP
-b
develop
cd
PaddleNLP
pip3
install
-e
.
```
<a
name=
"2"
></a>
## 2. 便捷使用
<a
name=
"21"
></a>
### 2.1 命令行使用
*
版面分析+表格识别
```
bash
paddleocr
--image_dir
=
../doc/table/1.png
--type
=
structure
```
*
VQA
coming soon
<a
name=
"22"
></a>
### 2.2 Python脚本使用
*
版面分析+表格识别
```
python
import
os
import
cv2
from
paddleocr
import
PPStructure
,
draw_structure_result
,
save_structure_res
table_engine
=
PPStructure
(
show_log
=
True
)
save_folder
=
'./output/table'
img_path
=
'../doc/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
=
'../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'
)
```
*
VQA
comming soon
<a
name=
"23"
></a>
### 2.3 返回结果说明
PP-Structure的返回结果为一个dict组成的list,示例如下
*
版面分析+表格识别
```
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 里各个字段说明如下
| 字段 | 说明 |
| --------------- | -------------|
|type|图片区域的类型|
|bbox|图片区域的在原图的坐标,分别[左上角x,左上角y,右下角x,右下角y]|
|res|图片区域的OCR或表格识别结果。
<br>
表格: 表格的HTML字符串;
<br>
OCR: 一个包含各个单行文字的检测坐标和识别结果的元组|
*
VQA
comming soon
<a
name=
"24"
></a>
### 2.4 参数说明
| 字段 | 说明 | 默认值 |
| --------------- | ---------------------------------------- | ------------------------------------------- |
| output | excel和识别结果保存的地址 | ./output/table |
| table_max_len | 表格结构模型预测时,图像的长边resize尺度 | 488 |
| table_model_dir | 表格结构模型 inference 模型地址 | None |
| table_char_type | 表格结构模型所用字典地址 | ../ppocr/utils/dict/table_structure_dict.txt |
| model_name_or_path | VQA SER模型地址 | None |
| max_seq_length | VQA SER模型最大支持token长度 | 512 |
| label_map_path | VQA SER 标签文件地址 | ./vqa/labels/labels_ser.txt |
| mode | pipeline预测模式,structure: 版面分析+表格识别; vqa: ser文档信息抽取 | structure |
大部分参数和paddleocr whl包保持一致,见
[
whl包文档
](
../doc/doc_ch/whl.md
)
运行完成后,每张图片会在
`output`
字段指定的目录下有一个同名目录,图片里的每个表格会存储为一个excel,图片区域会被裁剪之后保存下来,excel文件和图片名名为表格在图片里的坐标。
<a
name=
"3"
></a>
## 3. Python脚本使用
*
版面分析+表格识别
```
python
cd
ppstructure
# 下载模型
mkdir
inference
&&
cd
inference
# 下载超轻量级中文OCR模型的检测模型并解压
wget
https
:
//
paddleocr
.
bj
.
bcebos
.
com
/
dygraph_v2
.
0
/
ch
/
ch_ppocr_mobile_v2
.
0
_det_infer
.
tar
&&
tar
xf
ch_ppocr_mobile_v2
.
0
_det_infer
.
tar
# 下载超轻量级中文OCR模型的识别模型并解压
wget
https
:
//
paddleocr
.
bj
.
bcebos
.
com
/
dygraph_v2
.
0
/
ch
/
ch_ppocr_mobile_v2
.
0
_rec_infer
.
tar
&&
tar
xf
ch_ppocr_mobile_v2
.
0
_rec_infer
.
tar
# 下载超轻量级英文表格英寸模型并解压
wget
https
:
//
paddleocr
.
bj
.
bcebos
.
com
/
dygraph_v2
.
0
/
table
/
en_ppocr_mobile_v2
.
0
_table_structure_infer
.
tar
&&
tar
xf
en_ppocr_mobile_v2
.
0
_table_structure_infer
.
tar
cd
..
python3
predict_system
.
py
--
det_model_dir
=
inference
/
ch_ppocr_mobile_v2
.
0
_det_infer
\
--
rec_model_dir
=
inference
/
ch_ppocr_mobile_v2
.
0
_rec_infer
\
--
table_model_dir
=
inference
/
en_ppocr_mobile_v2
.
0
_table_structure_infer
\
--
image_dir
=
..
/
doc
/
table
/
1.
png
\
--
rec_char_dict_path
=
..
/
ppocr
/
utils
/
ppocr_keys_v1
.
txt
\
--
table_char_dict_path
=
..
/
ppocr
/
utils
/
dict
/
table_structure_dict
.
txt
\
--
output
=
..
/
output
/
table
\
--
vis_font_path
=
..
/
doc
/
fonts
/
simfang
.
ttf
```
运行完成后,每张图片会在
`output`
字段指定的目录下的
`talbe`
目录下有一个同名目录,图片里的每个表格会存储为一个excel,图片区域会被裁剪之后保存下来,excel文件和图片名名为表格在图片里的坐标。
*
VQA
```
python
cd
ppstructure
# 下载模型
mkdir
inference
&&
cd
inference
# 下载SER xfun 模型并解压
wget
https
:
//
paddleocr
.
bj
.
bcebos
.
com
/
pplayout
/
PP
-
Layout_v1
.
0
_ser_pretrained
.
tar
&&
tar
xf
PP
-
Layout_v1
.
0
_ser_pretrained
.
tar
cd
..
python3
predict_system
.
py
--
model_name_or_path
=
vqa
/
PP
-
Layout_v1
.
0
_ser_pretrained
/
\
--
mode
=
vqa
\
--
image_dir
=
vqa
/
images
/
input
/
zh_val_0
.
jpg
\
--
vis_font_path
=
..
/
doc
/
fonts
/
simfang
.
ttf
```
运行完成后,每张图片会在
`output`
字段指定的目录下的
`vqa`
目录下存放可视化之后的图片,图片名和输入图片名一致。
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录