2025-05-18 21:09:53 +08:00
|
|
|
|
---
|
|
|
|
|
comments: true
|
|
|
|
|
---
|
|
|
|
|
|
|
|
|
|
# 公式识别产线使用教程
|
|
|
|
|
|
|
|
|
|
## 1. 公式识别产线介绍
|
|
|
|
|
|
|
|
|
|
公式识别是一种自动从文档或图像中识别和提取LaTeX公式内容及其结构的技术,广泛应用于数学、物理、计算机科学等领域的文档编辑和数据分析。通过使用计算机视觉和机器学习算法,公式识别能够将复杂的数学公式信息转换为可编辑的LaTeX格式,方便用户进一步处理和分析数据。
|
|
|
|
|
|
|
|
|
|
公式识别产线用于解决公式识别任务,提取图片中的公式信息以LaTeX源码形式输出,本产线是一个集成了百度飞桨视觉团队自研的先进公式识别模型PP-FormulaNet 和业界知名公式识别模型 UniMERNet的端到端公式识别系统,支持简单印刷公式、复杂印刷公式、手写公式的识别,并在此基础上,增加了对图像的方向矫正和扭曲矫正功能。基于本产线,可实现公式内容精准预测,使用场景覆盖教育、科研、金融、制造等各个领域。本产线同时提供了灵活的服务化部署方式,支持在多种硬件上使用多种编程语言调用。不仅如此,本产线也提供了二次开发的能力,您可以基于本产线在您自己的数据集上训练调优,训练后的模型也可以无缝集成。
|
|
|
|
|
|
|
|
|
|
<img src="https://raw.githubusercontent.com/cuicheng01/PaddleX_doc_images/main/images/pipelines/formula_recognition/03.png" style="width: 70%"/>
|
|
|
|
|
|
|
|
|
|
<b>公式识别产线中包含以下4个模块。每个模块均可独立进行训练和推理,并包含多个模型。有关详细信息,请点击相应模块以查看文档。</b>
|
|
|
|
|
|
|
|
|
|
- [公式识别模块](../module_usage/formula_recognition.md)
|
|
|
|
|
- [版面区域检测模块](../module_usage/layout_detection.md)(可选)
|
|
|
|
|
- [文档图像方向分类模块](../module_usage/doc_img_orientation_classification.md) (可选)
|
|
|
|
|
- [文本图像矫正模块](../module_usage/text_image_unwarping.md) (可选)
|
|
|
|
|
|
|
|
|
|
在本产线中,您可以根据下方的基准测试数据选择使用的模型。
|
|
|
|
|
|
|
|
|
|
<details>
|
|
|
|
|
<summary><b>文档图像方向分类模块(可选):</b></summary>
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>模型</th><th>模型下载链接</th>
|
|
|
|
|
<th>Top-1 Acc(%)</th>
|
|
|
|
|
<th>GPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>CPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>模型存储大小(M)</th>
|
|
|
|
|
<th>介绍</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-LCNet_x1_0_doc_ori</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-LCNet_x1_0_doc_ori_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-LCNet_x1_0_doc_ori_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>99.06</td>
|
|
|
|
|
<td>2.31 / 0.43</td>
|
|
|
|
|
<td>3.37 / 1.27</td>
|
|
|
|
|
<td>7</td>
|
|
|
|
|
<td>基于PP-LCNet_x1_0的文档图像分类模型,含有四个类别,即0度,90度,180度,270度</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
</details>
|
|
|
|
|
|
|
|
|
|
<details>
|
|
|
|
|
<summary><b>文本图像矫正模块(可选):</b></summary>
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>模型</th><th>模型下载链接</th>
|
|
|
|
|
<th>CER </th>
|
|
|
|
|
<th>模型存储大小(M)</th>
|
|
|
|
|
<th>介绍</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>UVDoc</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/UVDoc_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/UVDoc_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>0.179</td>
|
|
|
|
|
<td>30.3 M</td>
|
|
|
|
|
<td>高精度文本图像矫正模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
</details>
|
|
|
|
|
|
|
|
|
|
<details>
|
|
|
|
|
<summary><b>版面区域检测模块(可选):</b></summary>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
|
|
|
|
|
* <b>版面检测模型,包含20个常见的类别:文档标题、段落标题、文本、页码、摘要、目录、参考文献、脚注、页眉、页脚、算法、公式、公式编号、图像、表格、图和表标题(图标题、表格标题和图表标题)、印章、图表、侧栏文本和参考文献内容</b>
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>模型</th><th>模型下载链接</th>
|
|
|
|
|
<th>mAP(0.5)(%)</th>
|
|
|
|
|
<th>GPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>CPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>模型存储大小(M)</th>
|
|
|
|
|
<th>介绍</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-DocLayout_plus-L</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout_plus-L_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-DocLayout_plus-L_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>83.2</td>
|
|
|
|
|
<td>34.6244 / 10.3945</td>
|
|
|
|
|
<td>510.57 / - </td>
|
|
|
|
|
<td>126.01 M</td>
|
|
|
|
|
<td>基于RT-DETR-L在包含中英文论文、多栏杂志、报纸、PPT、合同、书本、试卷、研报、古籍、日文文档、竖版文字文档等场景的自建数据集训练的更高精度版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
|
|
|
|
|
* <b>版面检测模型,包含23个常见的类别:文档标题、段落标题、文本、页码、摘要、目录、参考文献、脚注、页眉、页脚、算法、公式、公式编号、图像、图表标题、表格、表格标题、印章、图表标题、图表、页眉图像、页脚图像、侧栏文本</b>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>模型</th><th>模型下载链接</th>
|
|
|
|
|
<th>mAP(0.5)(%)</th>
|
|
|
|
|
<th>GPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>CPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>模型存储大小(M)</th>
|
|
|
|
|
<th>介绍</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-DocLayout-L</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout-L_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-DocLayout-L_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>90.4</td>
|
|
|
|
|
<td>34.6244 / 10.3945</td>
|
|
|
|
|
<td>510.57 / -</td>
|
|
|
|
|
<td>123.76 M</td>
|
|
|
|
|
<td>基于RT-DETR-L在包含中英文论文、杂志、合同、书本、试卷和研报等场景的自建数据集训练的高精度版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-DocLayout-M</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout-M_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-DocLayout-M_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>75.2</td>
|
|
|
|
|
<td>13.3259 / 4.8685</td>
|
|
|
|
|
<td>44.0680 / 44.0680</td>
|
|
|
|
|
<td>22.578</td>
|
|
|
|
|
<td>基于PicoDet-L在包含中英文论文、杂志、合同、书本、试卷和研报等场景的自建数据集训练的精度效率平衡的版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-DocLayout-S</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout-S_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-DocLayout-S_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>70.9</td>
|
|
|
|
|
<td>8.3008 / 2.3794</td>
|
|
|
|
|
<td>10.0623 / 9.9296</td>
|
|
|
|
|
<td>4.834</td>
|
|
|
|
|
<td>基于PicoDet-S在中英文论文、杂志、合同、书本、试卷和研报等场景上自建数据集训练的高效率版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
|
2025-05-20 00:28:19 +08:00
|
|
|
|
>❗ 以上列出的是版面区域检测模块重点支持的<b>4个核心模型</b>,该模块总共支持<b>7个全量模型</b>,包含多个预定义了不同类别的模型,完整的模型列表如下:
|
2025-05-18 21:09:53 +08:00
|
|
|
|
|
|
|
|
|
<details><summary> 👉模型列表详情</summary>
|
|
|
|
|
|
2025-05-20 00:28:19 +08:00
|
|
|
|
* <b>版面区域检测模型,包含17个版面常见类别,分别是:段落标题、图片、文本、数字、摘要、内容、图表标题、公式、表格、表格标题、参考文献、文档标题、脚注、页眉、算法、页脚、印章</b>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>模型</th><th>模型下载链接</th>
|
|
|
|
|
<th>mAP(0.5)(%)</th>
|
|
|
|
|
<th>GPU推理耗时(ms)</th>
|
|
|
|
|
<th>CPU推理耗时 (ms)</th>
|
|
|
|
|
<th>模型存储大小(M)</th>
|
|
|
|
|
<th>介绍</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PicoDet-S_layout_17cls</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet-S_layout_17cls_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PicoDet-S_layout_17cls_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>87.4</td>
|
|
|
|
|
<td>13.6</td>
|
|
|
|
|
<td>46.2</td>
|
|
|
|
|
<td>4.8</td>
|
|
|
|
|
<td>基于PicoDet-S轻量模型在中英文论文、杂志和研报等场景上自建数据集训练的高效率版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PicoDet-L_layout_17cls</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PicoDet-L_layout_17cls_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PicoDet-L_layout_17cls_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>89.0</td>
|
|
|
|
|
<td>17.2</td>
|
|
|
|
|
<td>160.2</td>
|
|
|
|
|
<td>22.6</td>
|
|
|
|
|
<td>基于PicoDet-L在中英文论文、杂志和研报等场景上自建数据集训练的效率精度均衡版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>RT-DETR-H_layout_17cls</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/RT-DETR-H_layout_17cls_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/RT-DETR-H_layout_17cls_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>98.3</td>
|
|
|
|
|
<td>115.1</td>
|
|
|
|
|
<td>3827.2</td>
|
|
|
|
|
<td>470.2</td>
|
|
|
|
|
<td>基于RT-DETR-H在中英文论文、杂志和研报等场景上自建数据集训练的高精度版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
|
|
|
|
|
|
2025-05-20 00:28:19 +08:00
|
|
|
|
* <b>版面区域检测模型,包含23个常见的类别:文档标题、段落标题、文本、页码、摘要、目录、参考文献、脚注、页眉、页脚、算法、公式、公式编号、图像、图表标题、表格、表格标题、印章、图表标题、图表、页眉图像、页脚图像、侧栏文本</b>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>模型</th><th>模型下载链接</th>
|
|
|
|
|
<th>mAP(0.5)(%)</th>
|
|
|
|
|
<th>GPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>CPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>模型存储大小(M)</th>
|
|
|
|
|
<th>介绍</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-DocLayout-L</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout-L_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-DocLayout-L_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>90.4</td>
|
|
|
|
|
<td>34.6244 / 10.3945</td>
|
|
|
|
|
<td>510.57 / -</td>
|
|
|
|
|
<td>123.76 M</td>
|
|
|
|
|
<td>基于RT-DETR-L在包含中英文论文、杂志、合同、书本、试卷和研报等场景的自建数据集训练的高精度版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-DocLayout-M</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout-M_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-DocLayout-M_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>75.2</td>
|
|
|
|
|
<td>13.3259 / 4.8685</td>
|
|
|
|
|
<td>44.0680 / 44.0680</td>
|
|
|
|
|
<td>22.578</td>
|
|
|
|
|
<td>基于PicoDet-L在包含中英文论文、杂志、合同、书本、试卷和研报等场景的自建数据集训练的精度效率平衡的版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-DocLayout-S</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout-S_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-DocLayout-S_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>70.9</td>
|
|
|
|
|
<td>8.3008 / 2.3794</td>
|
|
|
|
|
<td>10.0623 / 9.9296</td>
|
|
|
|
|
<td>4.834</td>
|
|
|
|
|
<td>基于PicoDet-S在中英文论文、杂志、合同、书本、试卷和研报等场景上自建数据集训练的高效率版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
|
|
|
|
|
* <b>版面检测模型,包含20个常见的类别:文档标题、段落标题、文本、页码、摘要、目录、参考文献、脚注、页眉、页脚、算法、公式、公式编号、图像、表格、图和表标题(图标题、表格标题和图表标题)、印章、图表、侧栏文本和参考文献内容</b>
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>模型</th><th>模型下载链接</th>
|
|
|
|
|
<th>mAP(0.5)(%)</th>
|
|
|
|
|
<th>GPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>CPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>模型存储大小(M)</th>
|
|
|
|
|
<th>介绍</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-DocLayout_plus-L</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-DocLayout_plus-L_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-DocLayout_plus-L_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>83.2</td>
|
|
|
|
|
<td>34.6244 / 10.3945</td>
|
|
|
|
|
<td>510.57 / - </td>
|
|
|
|
|
<td>126.01 M</td>
|
|
|
|
|
<td>基于RT-DETR-L在包含中英文论文、多栏杂志、报纸、PPT、合同、书本、试卷、研报、古籍、日文文档、竖版文字文档等场景的自建数据集训练的更高精度版面区域定位模型</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</details>
|
|
|
|
|
</details>
|
|
|
|
|
|
|
|
|
|
<details>
|
|
|
|
|
<summary><b>公式识别模块:</b></summary>
|
|
|
|
|
<table>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>模型</th><th>模型下载链接</th>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<th>En-BLEU(%)</th>
|
|
|
|
|
<th>Zh-BLEU(%)</th>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<th>GPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>CPU推理耗时(ms)<br/>[常规模式 / 高性能模式]</th>
|
|
|
|
|
<th>模型存储大小 (M)</th>
|
|
|
|
|
<th>介绍</th>
|
|
|
|
|
</tr>
|
|
|
|
|
<td>UniMERNet</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/UniMERNet_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/UniMERNet_pretrained.pdparams">训练模型</a></td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<td>85.91</td>
|
|
|
|
|
<td>43.50</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>2266.96/-</td>
|
|
|
|
|
<td>-/-</td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<td>1.53 G</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>UniMERNet是由上海AI Lab研发的一款公式识别模型。该模型采用Donut Swin作为编码器,MBartDecoder作为解码器,并通过在包含简单公式、复杂公式、扫描捕捉公式和手写公式在内的一百万数据集上进行训练,大幅提升了模型对真实场景公式的识别准确率</td>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-FormulaNet-S</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-FormulaNet-S_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-FormulaNet-S_pretrained.pdparams">训练模型</a></td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<td>87.00</td>
|
|
|
|
|
<td>45.71</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>202.25/-</td>
|
|
|
|
|
<td>-/-</td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<td>224 M</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td rowspan="2">PP-FormulaNet 是由百度飞桨视觉团队开发的一款先进的公式识别模型,支持5万个常见LateX源码词汇的识别。PP-FormulaNet-S 版本采用了 PP-HGNetV2-B4 作为其骨干网络,通过并行掩码和模型蒸馏等技术,大幅提升了模型的推理速度,同时保持了较高的识别精度,适用于简单印刷公式、跨行简单印刷公式等场景。而 PP-FormulaNet-L 版本则基于 Vary_VIT_B 作为骨干网络,并在大规模公式数据集上进行了深入训练,在复杂公式的识别方面,相较于PP-FormulaNet-S表现出显著的提升,适用于简单印刷公式、复杂印刷公式、手写公式等场景。 </td>
|
|
|
|
|
|
|
|
|
|
</tr>
|
|
|
|
|
<td>PP-FormulaNet-L</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-FormulaNet-L_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-FormulaNet-L_pretrained.pdparams">训练模型</a></td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<td>90.36</td>
|
|
|
|
|
<td>45.78</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>1976.52/-</td>
|
|
|
|
|
<td>-/-</td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<td>695 M</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<tr>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<td>PP-FormulaNet_plus-S</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-FormulaNet_plus-S_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-FormulaNet_plus-S_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>88.71</td>
|
|
|
|
|
<td>53.32</td>
|
|
|
|
|
<td>191.69/-</td>
|
|
|
|
|
<td>-/-</td>
|
|
|
|
|
<td>248 M</td>
|
|
|
|
|
<td rowspan="3">PP-FormulaNet_plus 是百度飞桨视觉团队在 PP-FormulaNet 的基础上开发的增强版公式识别模型。与原版相比,PP-FormulaNet_plus 在训练中使用了更为丰富的公式数据集,包括中文学位论文、专业书籍、教材试卷以及数学期刊等多种来源。这一扩展显著提升了模型的识别能力。
|
|
|
|
|
|
|
|
|
|
其中,PP-FormulaNet_plus-M 和 PP-FormulaNet_plus-L 模型新增了对中文公式的支持,并将公式的最大预测 token 数从 1024 扩大至 2560,大幅提升了对复杂公式的识别性能。同时,PP-FormulaNet_plus-S 模型则专注于增强英文公式的识别能力。通过这些改进,PP-FormulaNet_plus 系列模型在处理复杂多样的公式识别任务时表现更加出色。 </td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-FormulaNet_plus-M</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-FormulaNet_plus-M_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-FormulaNet_plus-M_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>91.45</td>
|
|
|
|
|
<td>89.76</td>
|
|
|
|
|
<td>1301.56/-</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>-/-</td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<td>592 M</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>PP-FormulaNet_plus-L</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/PP-FormulaNet_plus-L_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/PP-FormulaNet_plus-L_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>92.22</td>
|
|
|
|
|
<td>90.64</td>
|
|
|
|
|
<td>1745.25/-</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>-/-</td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<td>698 M</td>
|
|
|
|
|
</tr>
|
|
|
|
|
|
|
|
|
|
<tr>
|
|
|
|
|
<td>LaTeX_OCR_rec</td><td><a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_inference_model/paddle3.0.0/LaTeX_OCR_rec_infer.tar">推理模型</a>/<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/official_pretrained_model/LaTeX_OCR_rec_pretrained.pdparams">训练模型</a></td>
|
|
|
|
|
<td>74.55</td>
|
|
|
|
|
<td>39.96</td>
|
|
|
|
|
<td>1244.61/-</td>
|
|
|
|
|
<td>-/-</td>
|
|
|
|
|
<td>99 M</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>LaTeX-OCR是一种基于自回归大模型的公式识别算法,通过采用 Hybrid ViT 作为骨干网络,transformer作为解码器,显著提升了公式识别的准确性。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</table>
|
|
|
|
|
</details>
|
|
|
|
|
|
|
|
|
|
<details>
|
|
|
|
|
<summary><b>测试环境说明:</b></summary>
|
|
|
|
|
|
|
|
|
|
<ul>
|
|
|
|
|
<li><b>性能测试环境</b>
|
|
|
|
|
<ul>
|
|
|
|
|
<li><strong>测试数据集:
|
|
|
|
|
</strong>
|
|
|
|
|
<ul>
|
2025-05-20 17:42:04 +08:00
|
|
|
|
<li>文档图像方向分类模型:自建的内部数据集,覆盖证件和文档等多个场景,包含 1000 张图片。</li>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<li> 文本图像矫正模型:<a href="https://www3.cs.stonybrook.edu/~cvl/docunet.html">DocUNet</a>。</li>
|
|
|
|
|
<li>版面区域检测模型:PaddleOCR 自建的版面区域检测数据集,包含中英文论文、杂志、合同、书本、试卷和研报等常见的 500 张文档类型图片。</li>
|
|
|
|
|
<li>17类区域检测模型:PaddleOCR 自建的版面区域检测数据集,包含中英文论文、杂志和研报等常见的 892 张文档类型图片。</li>
|
2025-05-20 17:42:04 +08:00
|
|
|
|
<li>公式识别模型:自建的内部公式识别测试集。</li>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</ul>
|
|
|
|
|
</li>
|
|
|
|
|
<li><strong>硬件配置:</strong>
|
|
|
|
|
<ul>
|
|
|
|
|
<li>GPU:NVIDIA Tesla T4</li>
|
|
|
|
|
<li>CPU:Intel Xeon Gold 6271C @ 2.60GHz</li>
|
|
|
|
|
<li>其他环境:Ubuntu 20.04 / cuDNN 8.6 / TensorRT 8.5.2.2</li>
|
|
|
|
|
</ul>
|
|
|
|
|
</li>
|
|
|
|
|
</ul>
|
|
|
|
|
</li>
|
|
|
|
|
<li><b>推理模式说明</b></li>
|
|
|
|
|
</ul>
|
|
|
|
|
|
|
|
|
|
<table border="1">
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>模式</th>
|
|
|
|
|
<th>GPU配置</th>
|
|
|
|
|
<th>CPU配置</th>
|
|
|
|
|
<th>加速技术组合</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>常规模式</td>
|
|
|
|
|
<td>FP32精度 / 无TRT加速</td>
|
|
|
|
|
<td>FP32精度 / 8线程</td>
|
|
|
|
|
<td>PaddleInference</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>高性能模式</td>
|
|
|
|
|
<td>选择先验精度类型和加速策略的最优组合</td>
|
|
|
|
|
<td>FP32精度 / 8线程</td>
|
|
|
|
|
<td>选择先验最优后端(Paddle/OpenVINO/TRT等)</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
</details>
|
|
|
|
|
|
|
|
|
|
<br />
|
|
|
|
|
<b>如果您更注重模型的精度,请选择精度较高的模型;如果您更在意模型的推理速度,请选择推理速度较快的模型;如果您关注模型的存储大小,请选择存储体积较小的模型。</b>
|
|
|
|
|
|
|
|
|
|
## 2. 快速开始
|
|
|
|
|
|
2025-05-20 19:29:01 +08:00
|
|
|
|
在本地使用公式识别产线前,请确保您已经按照[安装教程](../installation.md)完成了wheel包安装。安装完成后,可以在本地使用命令行体验或 Python 集成。
|
2025-05-18 21:09:53 +08:00
|
|
|
|
|
|
|
|
|
### 2.1 命令行方式体验
|
|
|
|
|
|
2025-05-27 20:14:53 +08:00
|
|
|
|
一行命令即可快速体验 formula_recognition 产线效果。运行以下代码前,请您下载[示例图片](https://paddle-model-ecology.bj.bcebos.com/paddlex/demo_image/pipelines/general_formula_recognition_001.png)到本地:
|
2025-05-18 21:09:53 +08:00
|
|
|
|
|
|
|
|
|
```bash
|
|
|
|
|
paddleocr formula_recognition_pipeline -i https://paddle-model-ecology.bj.bcebos.com/paddlex/demo_image/pipelines/general_formula_recognition_001.png
|
|
|
|
|
|
|
|
|
|
# 通过 --use_doc_orientation_classify 指定是否使用文档方向分类模型
|
|
|
|
|
paddleocr formula_recognition_pipeline -i ./general_formula_recognition_001.png --use_doc_orientation_classify True
|
|
|
|
|
|
|
|
|
|
# 通过 --use_doc_unwarping 指定是否使用文本图像矫正模块
|
|
|
|
|
paddleocr formula_recognition_pipeline -i ./general_formula_recognition_001.png --use_doc_unwarping True
|
|
|
|
|
|
|
|
|
|
# 通过 --device 指定模型推理时使用 GPU
|
|
|
|
|
paddleocr formula_recognition_pipeline -i ./general_formula_recognition_001.png --device gpu
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
<details><summary><b>命令行支持更多参数设置,点击展开以查看命令行参数的详细说明</b></summary>
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>参数</th>
|
|
|
|
|
<th>参数说明</th>
|
|
|
|
|
<th>参数类型</th>
|
|
|
|
|
<th>默认值</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
<td><code>input</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>待预测数据,必填。
|
|
|
|
|
如图像文件或者PDF文件的本地路径:<code>/root/data/img.jpg</code>;<b>如URL链接</b>,如图像文件或PDF文件的网络URL:<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/demo_image/pipelines/general_formula_recognition_001.png">示例</a>;<b>如本地目录</b>,该目录下需包含待预测图像,如本地路径:<code>/root/data/</code>(当前不支持目录中包含PDF文件的预测,PDF文件需要指定到具体文件路径)。
|
2025-05-20 15:17:30 +08:00
|
|
|
|
</td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td></td>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>save_path</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>指定推理结果文件保存的路径。如果不设置,推理结果将不会保存到本地。</td>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>doc_orientation_classify_model_name</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文档方向分类模型的名称。如果不设置,将会使用产线默认模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_orientation_classify_model_dir</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文档方向分类模型的目录路径。如果不设置,将会下载官方模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_orientation_classify_batch_size</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文档方向分类模型的批处理大小。如果不设置,将默认设置批处理大小为<code>1</code>。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>int</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_unwarping_model_name</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文本图像矫正模型的名称。如果不设置,将会使用产线默认模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_unwarping_model_dir</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文本图像矫正模型的目录路径。如果不设置,将会下载官方模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_unwarping_batch_size</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文本图像矫正模型的批处理大小。如果不设置,将默认设置批处理大小为<code>1</code>。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>int</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
<td><code>use_doc_orientation_classify</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>是否加载并使用文档方向分类模块。如果不设置,将默认使用产线初始化的该参数值,初始化为<code>True</code>。</td>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
<td><code>bool</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>use_doc_unwarping</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>是否加载并使用文本图像矫正模块。如果不设置,将默认使用产线初始化的该参数值,初始化为<code>True</code>。</td>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
<td><code>bool</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<tr>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>layout_detection_model_name</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>版面区域检测模型的名称。如果不设置,将会使用产线默认模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_detection_model_dir</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>版面区域检测模型的目录路径。如果不设置,将会下载官方模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_threshold</code></td>
|
|
|
|
|
<td>版面区域检测的阈值,用于过滤掉低置信度预测结果的阈值。
|
2025-05-28 10:24:37 +08:00
|
|
|
|
如 0.2,表示过滤掉所有阈值小于0.2的目标框。如果不设置,将默认使用默认值。
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td><code>float</code></td>
|
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_nms</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>版面区域检测是否使用NMS后处理,过滤重叠框。如果不设置,将会使用官方模型配置。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>bool</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_unclip_ratio</code></td>
|
|
|
|
|
<td>版面区域检测中检测框的边长缩放倍数。
|
2025-05-28 10:24:37 +08:00
|
|
|
|
大于0的浮点数,如 1.1 ,表示将模型输出的检测框中心不变,宽和高都扩张1.1倍。如果不设置,将使用默认值:1.0。
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td><code>float</code></td>
|
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_merge_bboxes_mode</code></td>
|
|
|
|
|
<td>版面区域检测中模型输出的检测框的合并处理模式。
|
|
|
|
|
<ul>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<li><b>large</b>,设置为large时,表示在模型输出的检测框中,对于互相重叠包含的检测框,只保留外部最大的框,删除重叠的内部框;</li>
|
|
|
|
|
<li><b>small</b>,设置为small,表示在模型输出的检测框中,对于互相重叠包含的检测框,只保留内部被包含的小框,删除重叠的外部框;</li>
|
|
|
|
|
<li><b>union</b>,不进行框的过滤处理,内外框都保留</li>
|
|
|
|
|
</ul>如果不设置,将使用默认值:“large”;
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</td>
|
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_detection_batch_size</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>版面区域检测模型的批处理大小。如果不设置,将默认设置批处理大小为<code>1</code>。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>int</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<td><code>use_layout_detection</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>是否加载并使用版面区域检测模块。如果不设置,将默认使用产线初始化的该参数值,初始化为<code>True</code>。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>bool</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>formula_recognition_model_name</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>公式识别模型的名称。如果不设置,将会使用产线默认模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>formula_recognition_model_dir</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>公式识别模型的目录路径。如果不设置,将会下载官方模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>formula_recognition_batch_size</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>公式识别模型的批处理大小。如果不设置,将默认设置批处理大小为<code>1</code>。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>int</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>device</code></td>
|
|
|
|
|
<td>用于推理的设备。支持指定具体卡号。
|
|
|
|
|
<ul>
|
|
|
|
|
<li><b>CPU</b>:如 <code>cpu</code> 表示使用 CPU 进行推理;</li>
|
|
|
|
|
<li><b>GPU</b>:如 <code>gpu:0</code> 表示使用第 1 块 GPU 进行推理;</li>
|
|
|
|
|
<li><b>NPU</b>:如 <code>npu:0</code> 表示使用第 1 块 NPU 进行推理;</li>
|
|
|
|
|
<li><b>XPU</b>:如 <code>xpu:0</code> 表示使用第 1 块 XPU 进行推理;</li>
|
|
|
|
|
<li><b>MLU</b>:如 <code>mlu:0</code> 表示使用第 1 块 MLU 进行推理;</li>
|
|
|
|
|
<li><b>DCU</b>:如 <code>dcu:0</code> 表示使用第 1 块 DCU 进行推理;</li>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
</ul>如果不设置, 将默认使用产线初始化的该参数值,初始化时,会优先使用本地的 GPU 0号设备,如果没有,则使用 CPU 设备。
|
|
|
|
|
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</td>
|
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>enable_hpi</code></td>
|
|
|
|
|
<td>是否启用高性能推理。</td>
|
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>False</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>use_tensorrt</code></td>
|
|
|
|
|
<td>是否使用 TensorRT 进行推理加速。</td>
|
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>False</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>min_subgraph_size</code></td>
|
|
|
|
|
<td>最小子图大小,用于优化模型子图的计算。</td>
|
|
|
|
|
<td><code>int</code></td>
|
|
|
|
|
<td><code>3</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>precision</code></td>
|
|
|
|
|
<td>计算精度,如 fp32、fp16。</td>
|
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>fp32</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>enable_mkldnn</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>是否启用 MKL-DNN 加速库。如果不设置,将默认启用。
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</td>
|
|
|
|
|
<td><code>bool</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>cpu_threads</code></td>
|
|
|
|
|
<td>在 CPU 上进行推理时使用的线程数。</td>
|
|
|
|
|
<td><code>int</code></td>
|
|
|
|
|
<td><code>8</code></td>
|
|
|
|
|
</tr>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
<tr>
|
|
|
|
|
<td><code>paddlex_config</code></td>
|
|
|
|
|
<td>PaddleX产线配置文件路径。</td>
|
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
</tr>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
</details>
|
|
|
|
|
<br />
|
|
|
|
|
|
|
|
|
|
运行结果会被打印到终端上,默认配置的 formula_recognition 产线的运行结果如下:
|
|
|
|
|
|
|
|
|
|
```bash
|
2025-05-20 00:28:19 +08:00
|
|
|
|
{'res': {'input_path': './general_formula_recognition_001.png', 'page_index': None, 'model_settings': {'use_doc_preprocessor': True, 'use_layout_detection': True}, 'doc_preprocessor_res': {'input_path': None, 'page_index': None, 'model_settings': {'use_doc_orientation_classify': True, 'use_doc_unwarping': True}, 'angle': 0}, 'layout_det_res': {'input_path': None, 'page_index': None, 'boxes': [{'cls_id': 2, 'label': 'text', 'score': 0.9855189323425293, 'coordinate': [90.56131, 1086.7773, 658.8992, 1553.2681]}, {'cls_id': 2, 'label': 'text', 'score': 0.9814704060554504, 'coordinate': [93.04651, 127.988556, 664.8587, 396.60892]}, {'cls_id': 2, 'label': 'text', 'score': 0.9767388105392456, 'coordinate': [698.4391, 591.0454, 1293.3676, 748.28345]}, {'cls_id': 2, 'label': 'text', 'score': 0.9712911248207092, 'coordinate': [701.4946, 286.61566, 1299.0099, 391.87457]}, {'cls_id': 2, 'label': 'text', 'score': 0.9709068536758423, 'coordinate': [697.0126, 751.93604, 1290.2236, 883.64453]}, {'cls_id': 2, 'label': 'text', 'score': 0.9689271450042725, 'coordinate': [704.01196, 79.645935, 1304.7493, 187.96674]}, {'cls_id': 2, 'label': 'text', 'score': 0.9683637619018555, 'coordinate': [93.063385, 799.3567, 660.6935, 902.0344]}, {'cls_id': 7, 'label': 'formula', 'score': 0.9660536646842957, 'coordinate': [728.5045, 440.9215, 1224.0634, 570.8518]}, {'cls_id': 7, 'label': 'formula', 'score': 0.9616329669952393, 'coordinate': [722.9789, 1333.5085, 1257.1136, 1468.0432]}, {'cls_id': 7, 'label': 'formula', 'score': 0.9610316753387451, 'coordinate': [756.4525, 1211.323, 1188.0428, 1268.2336]}, {'cls_id': 7, 'label': 'formula', 'score': 0.960993230342865, 'coordinate': [777.51355, 207.87927, 1222.8966, 267.33014]}, {'cls_id': 2, 'label': 'text', 'score': 0.9594196677207947, 'coordinate': [697.5154, 957.6764, 1288.6238, 1033.5211]}, {'cls_id': 2, 'label': 'text', 'score': 0.9593432545661926, 'coordinate': [691.333, 1511.8015, 1282.0968, 1642.5906]}, {'cls_id': 7, 'label': 'formula', 'score': 0.9589930176734924, 'coordinate': [153.89856, 924.2046, 601.0946, 1036.9038]}, {'cls_id': 2, 'label': 'text', 'score': 0.9582098722457886, 'coordinate': [87.02347, 1557.2971, 655.9584, 1632.6912]}, {'cls_id': 7, 'label': 'formula', 'score': 0.9579620957374573, 'coordinate': [810.86975, 1057.0771, 1175.101, 1117.6631]}, {'cls_id': 7, 'label': 'formula', 'score': 0.9557801485061646, 'coordinate': [165.26271, 557.8495, 598.1803, 614.35]}, {'cls_id': 7, 'label': 'formula', 'score': 0.953873872756958, 'coordinate': [116.48187, 713.88416, 614.2181, 774.02576]}, {'cls_id': 2, 'label': 'text', 'score': 0.9521227478981018, 'coordinate': [96.6882, 478.32745, 662.573, 536.5877]}, {'cls_id': 2, 'label': 'text', 'score': 0.944242000579834, 'coordinate': [96.12866, 639.1591, 661.7959, 692.4849]}, {'cls_id': 2, 'label': 'text', 'score': 0.9403323531150818, 'coordinate': [695.9436, 1138.6748, 1286.7242, 1188.0049]}, {'cls_id': 7, 'label': 'formula', 'score': 0.9249663949012756, 'coordinate': [852.90137, 908.64386, 1131.1882, 933.81793]}, {'cls_id': 7, 'label': 'formula', 'score': 0.9249223470687866, 'coordinate': [195.28397, 424.81024, 567.697, 451.1291]}, {'cls_id': 17, 'label': 'formula_number', 'score': 0.9173304438591003, 'coordinate': [1246.2393, 1079.0535, 1286.3281, 1104.3323]}, {'cls_id': 17, 'label': 'formula_number', 'score': 0.9169507026672363, 'coordinate': [1246.9003, 908.6482, 1288.2013, 934.61426]}, {'cls_id': 17, 'label': 'formula_number', 'score': 0.915979266166687, 'coordinate': [1247.0374, 1229.1572, 1287.094, 1254.9805]}, {'cls_id': 17, 'label': 'formula_number', 'score': 0.9085646867752075, 'coordinate': [1252.864, 492.1079, 1294.6238, 518.47095]}, {'cls_id': 17, 'label': 'formula_number', 'score': 0.9017605781555176, 'coordinate': [1242.1719, 1473.6951, 1283.02, 1498.6316]}, {'cls_id': 17, 'label': 'formula_number', 'score': 0.8999755382537842, 'coordinate': [1269.8164, 220.34933, 1299.8589, 247.01102]}, {'cls_id': 7, 'label': 'formula', 'score': 0.8965252041816711, 'coordinate': [96.00711, 235.49493, 295.43823, 265.60016]}, {'cls_id': 2, 'label': 't
|
2025-05-18 21:09:53 +08:00
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
运行结果参数说明可以参考[2.2 Python脚本方式集成](#22-python脚本方式集成)中的结果解释。
|
|
|
|
|
|
|
|
|
|
可视化结果保存在`save_path`下,其中公式识别的可视化结果如下:
|
|
|
|
|
|
2025-05-20 00:28:19 +08:00
|
|
|
|
<img src="https://raw.githubusercontent.com/cuicheng01/PaddleX_doc_images/main/images/pipelines/formula_recognition/04_paddleocr3.png" style="width: 70%"/>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
|
|
|
|
|
<b> 如果您需要对公式识别产线进行可视化,需要运行如下命令来对LaTeX渲染环境进行安装。目前公式识别产线可视化只支持Ubuntu环境,其他环境暂不支持。对于复杂公式,LaTeX 结果可能包含部分高级的表示,Markdown等环境中未必可以成功显示:</b>
|
|
|
|
|
|
|
|
|
|
```bash
|
|
|
|
|
sudo apt-get update
|
2025-05-20 00:28:19 +08:00
|
|
|
|
sudo apt-get install texlive texlive-latex-base texlive-xetex latex-cjk-all texlive-latex-extra -y
|
2025-05-18 21:09:53 +08:00
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
<b>备注</b>: 由于公式识别可视化过程中需要对每张公式图片进行渲染,因此耗时较长,请您耐心等待。
|
|
|
|
|
|
|
|
|
|
### 2.2 Python脚本方式集成
|
|
|
|
|
|
|
|
|
|
命令行方式是为了快速体验查看效果,一般来说,在项目中,往往需要通过代码集成,您可以通过几行代码即可完成产线的快速推理,推理代码如下:
|
|
|
|
|
|
|
|
|
|
```python
|
|
|
|
|
from paddleocr import FormulaRecognitionPipeline
|
|
|
|
|
|
|
|
|
|
pipeline = FormulaRecognitionPipeline()
|
|
|
|
|
# ocr = FormulaRecognitionPipeline(use_doc_orientation_classify=True) # 通过 use_doc_orientation_classify 指定是否使用文档方向分类模型
|
|
|
|
|
# ocr = FormulaRecognitionPipeline(use_doc_unwarping=True) # 通过 use_doc_unwarping 指定是否使用文本图像矫正模块
|
|
|
|
|
# ocr = FormulaRecognitionPipeline(device="gpu") # 通过 device 指定模型推理时使用 GPU
|
|
|
|
|
output = pipeline.predict("./general_formula_recognition_001.png")
|
|
|
|
|
for res in output:
|
|
|
|
|
res.print() ## 打印预测的结构化输出
|
2025-05-20 00:28:19 +08:00
|
|
|
|
res.save_to_img(save_path="output") ## 保存当前图像的公式可视化结果
|
2025-05-18 21:09:53 +08:00
|
|
|
|
res.save_to_json(save_path="output") ## 保存当前图像的结构化json结果
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
在上述 Python 脚本中,执行了如下几个步骤:
|
|
|
|
|
|
|
|
|
|
(1)通过 `FormulaRecognitionPipeline()` 实例化公式识别产线对象,具体参数说明如下:
|
|
|
|
|
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>参数</th>
|
|
|
|
|
<th>参数说明</th>
|
|
|
|
|
<th>参数类型</th>
|
|
|
|
|
<th>默认值</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_orientation_classify_model_name</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文档方向分类模型的名称。如果设置为<code>None</code>,将会使用产线默认模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_orientation_classify_model_dir</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文档方向分类模型的目录路径。如果设置为<code>None</code>,将会下载官方模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_orientation_classify_batch_size</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文档方向分类模型的批处理大小。如果设置为<code>None</code>,将默认设置批处理大小为<code>1</code>。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>int</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_unwarping_model_name</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文本图像矫正模型的名称。如果设置为<code>None</code>,将会使用产线默认模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_unwarping_model_dir</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文本图像矫正模型的目录路径。如果设置为<code>None</code>,将会下载官方模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>doc_unwarping_batch_size</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>文本图像矫正模型的批处理大小。如果设置为<code>None</code>,将默认设置批处理大小为<code>1</code>。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>int</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
<td><code>use_doc_orientation_classify</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>是否加载并使用文档方向分类模块。如果设置为<code>None</code>,将默认使用产线初始化的该参数值,初始化为<code>True</code>。</td>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>use_doc_unwarping</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>是否加载并使用文本图像矫正模块。如果设置为<code>None</code>,将默认使用产线初始化的该参数值,初始化为<code>True</code>。</td>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>layout_detection_model_name</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>版面区域检测模型的名称。如果设置为<code>None</code>,将会使用产线默认模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_detection_model_dir</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>版面区域检测模型的目录路径。如果设置为<code>None</code>,将会下载官方模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_threshold</code></td>
|
|
|
|
|
<td>版面区域检测的阈值,用于过滤掉低置信度预测结果的阈值。
|
|
|
|
|
<ul>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<li><b>float</b>:如 0.2, 表示过滤掉所有阈值小于0.2的目标框;</li>
|
|
|
|
|
<li><b>字典</b>:字典的key为<b>int</b>类型,代表<code>cls_id</code>,val为<b>float</b>类型阈值。如 <code>{0: 0.45,2: 0.48,7: 0.4}</code>,表示对cls_id为0的类别应用阈值0.45、cls_id为2的类别应用阈值0.48、cls_id为7的类别应用阈值0.4;</li>
|
|
|
|
|
<li><b>None</b>:不指定,将使用默认值:0.5。</li>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</td>
|
|
|
|
|
<td><code>float|dict</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_nms</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>版面区域检测是否使用NMS后处理,过滤重叠框。如果设置为<code>None</code>,将会使用官方模型配置。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_unclip_ratio</code></td>
|
|
|
|
|
<td>版面区域检测中检测框的边长缩放倍数。
|
|
|
|
|
<ul>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<li><b>float</b>:大于0的浮点数,如 1.1,表示将模型输出的检测框中心不变,宽和高都扩张1.1倍;</li>
|
|
|
|
|
<li><b>列表</b>:如 [1.2,1.5] ,表示将模型输出的检测框中心不变,宽度扩张1.2倍,高度扩张1.5倍;</li>
|
|
|
|
|
<li><b>None</b>:不指定,将使用默认值:1.0。</li>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</ul>
|
|
|
|
|
</td>
|
|
|
|
|
<td><code>float|list</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_merge_bboxes_mode</code></td>
|
|
|
|
|
<td>版面区域检测中模型输出的检测框的合并处理模式。
|
|
|
|
|
<ul>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<li><b>large</b>:设置为large时,表示在模型输出的检测框中,对于互相重叠包含的检测框,只保留外部最大的框,删除重叠的内部框;</li>
|
|
|
|
|
<li><b>small</b>:设置为small,表示在模型输出的检测框中,对于互相重叠包含的检测框,只保留内部被包含的小框,删除重叠的外部框;</li>
|
|
|
|
|
<li><b>union</b>:不进行框的过滤处理,内外框都保留;</li>
|
|
|
|
|
<li><b>None</b>:不指定,将使用默认值:“large”。</li>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</ul>
|
|
|
|
|
</td>
|
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_detection_batch_size</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>版面区域检测模型的批处理大小。如果设置为<code>None</code>,将默认设置批处理大小为<code>1</code>。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>int</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>use_layout_detection</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>是否加载并使用版面区域检测模块。如果设置为<code>None</code>,将默认使用产线初始化的该参数值,初始化为<code>True</code>。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>formula_recognition_model_name</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>公式识别模型的名称。如果设置为<code>None</code>,将会使用产线默认模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>formula_recognition_model_dir</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>公式识别模型的目录路径。如果设置为<code>None</code>,将会下载官方模型。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>formula_recognition_batch_size</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>公式识别模型的批处理大小。如果设置为<code>None</code>,将默认设置批处理大小为<code>1</code>。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>int</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>device</code></td>
|
|
|
|
|
<td>用于推理的设备。支持指定具体卡号。
|
|
|
|
|
<ul>
|
|
|
|
|
<li><b>CPU</b>:如 <code>cpu</code> 表示使用 CPU 进行推理;</li>
|
|
|
|
|
<li><b>GPU</b>:如 <code>gpu:0</code> 表示使用第 1 块 GPU 进行推理;</li>
|
|
|
|
|
<li><b>NPU</b>:如 <code>npu:0</code> 表示使用第 1 块 NPU 进行推理;</li>
|
|
|
|
|
<li><b>XPU</b>:如 <code>xpu:0</code> 表示使用第 1 块 XPU 进行推理;</li>
|
|
|
|
|
<li><b>MLU</b>:如 <code>mlu:0</code> 表示使用第 1 块 MLU 进行推理;</li>
|
|
|
|
|
<li><b>DCU</b>:如 <code>dcu:0</code> 表示使用第 1 块 DCU 进行推理;</li>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<li><b>None</b>:如果设置为<code>None</code>,将默认使用产线初始化的该参数值,初始化时,会优先使用本地的 GPU 0号设备,如果没有,则使用 CPU 设备。</li>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</ul>
|
|
|
|
|
</td>
|
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>enable_hpi</code></td>
|
|
|
|
|
<td>是否启用高性能推理。</td>
|
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>False</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>use_tensorrt</code></td>
|
|
|
|
|
<td>是否使用 TensorRT 进行推理加速。</td>
|
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>False</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>min_subgraph_size</code></td>
|
|
|
|
|
<td>最小子图大小,用于优化模型子图的计算。</td>
|
|
|
|
|
<td><code>int</code></td>
|
|
|
|
|
<td><code>3</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>precision</code></td>
|
|
|
|
|
<td>计算精度,如 fp32、fp16。</td>
|
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td><code>"fp32"</code></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>enable_mkldnn</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>是否启用 MKL-DNN 加速库。如果设置为<code>None</code>,将默认启用。
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</td>
|
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>cpu_threads</code></td>
|
|
|
|
|
<td>在 CPU 上进行推理时使用的线程数。</td>
|
|
|
|
|
<td><code>int</code></td>
|
|
|
|
|
<td><code>8</code></td>
|
|
|
|
|
</tr>
|
2025-05-20 15:17:30 +08:00
|
|
|
|
<tr>
|
|
|
|
|
<td><code>paddlex_config</code></td>
|
|
|
|
|
<td>PaddleX产线配置文件路径。</td>
|
|
|
|
|
<td><code>str</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
|
|
|
|
|
(2)调用 公式识别产线对象的 `predict()` 方法进行推理预测,该方法会返回一个结果列表。
|
|
|
|
|
|
|
|
|
|
另外,产线还提供了 `predict_iter()` 方法。两者在参数接受和结果返回方面是完全一致的,区别在于 `predict_iter()` 返回的是一个 `generator`,能够逐步处理和获取预测结果,适合处理大型数据集或希望节省内存的场景。可以根据实际需求选择使用这两种方法中的任意一种。
|
|
|
|
|
|
|
|
|
|
以下是 `predict()` 方法的参数及其说明:
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>参数</th>
|
|
|
|
|
<th>参数说明</th>
|
|
|
|
|
<th>参数类型</th>
|
|
|
|
|
<th>默认值</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>input</code></td>
|
|
|
|
|
<td>待预测数据,支持多种输入类型,必填
|
|
|
|
|
<ul>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<li><b>Python Var</b>:如 <code>numpy.ndarray</code> 表示的图像数据;</li>
|
|
|
|
|
<li><b>str</b>:如图像文件或者PDF文件的本地路径:<code>/root/data/img.jpg</code>;<b>如URL链接</b>,如图像文件或PDF文件的网络URL:<a href="https://paddle-model-ecology.bj.bcebos.com/paddlex/demo_image/pipelines/general_formula_recognition_001.png">示例</a>;<b>如本地目录</b>,该目录下需包含待预测图像,如本地路径:<code>/root/data/</code>(当前不支持目录中包含PDF文件的预测,PDF文件需要指定到具体文件路径);</li>
|
|
|
|
|
<li><b>List</b>:列表元素需为上述类型数据,如<code>[numpy.ndarray, numpy.ndarray]</code>,<code>["/root/data/img1.jpg", "/root/data/img2.jpg"]</code>,<code>["/root/data1", "/root/data2"]。</code></li>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</ul>
|
|
|
|
|
</td>
|
|
|
|
|
<td><code>Python Var|str|list</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td></td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>use_layout_detection</code></td>
|
|
|
|
|
<td>是否在推理时使用文档区域检测模块。</td>
|
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>use_doc_orientation_classify</code></td>
|
|
|
|
|
<td>是否在推理时使用文档方向分类模块。</td>
|
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>use_doc_unwarping</code></td>
|
|
|
|
|
<td>是否在推理时使用文本图像矫正模块。</td>
|
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_threshold</code></td>
|
|
|
|
|
<td>与实例化时的参数相同。</td>
|
|
|
|
|
<td><code>float|dict</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_nms</code></td>
|
|
|
|
|
<td>与实例化时的参数相同。</td>
|
|
|
|
|
<td><code>bool</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_unclip_ratio</code></td>
|
|
|
|
|
<td>与实例化时的参数相同。</td>
|
|
|
|
|
<td><code>float|list</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layout_merge_bboxes_mode</code></td>
|
|
|
|
|
<td>与实例化时的参数相同。</td>
|
|
|
|
|
<td><code>string</code></td>
|
|
|
|
|
<td><code>None</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tr></tr></tbody>
|
|
|
|
|
</table>
|
|
|
|
|
|
|
|
|
|
(3)对预测结果进行处理,每个样本的预测结果均为对应的Result对象,且支持打印、保存为图片、保存为`json`文件的操作:
|
|
|
|
|
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>方法</th>
|
|
|
|
|
<th>方法说明</th>
|
|
|
|
|
<th>参数</th>
|
|
|
|
|
<th>参数类型</th>
|
|
|
|
|
<th>参数说明</th>
|
|
|
|
|
<th>默认值</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<td rowspan="3"><code>print()</code></td>
|
|
|
|
|
<td rowspan="3">打印结果到终端</td>
|
|
|
|
|
<td><code>format_json</code></td>
|
|
|
|
|
<td><code>bool</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>是否对输出内容进行使用 <code>JSON</code> 缩进格式化。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>True</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>indent</code></td>
|
|
|
|
|
<td><code>int</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>指定缩进级别,以美化输出的 <code>JSON</code> 数据,使其更具可读性,仅当 <code>format_json</code> 为 <code>True</code> 时有效。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>4</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>ensure_ascii</code></td>
|
|
|
|
|
<td><code>bool</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>控制是否将非 <code>ASCII</code> 字符转义为 <code>Unicode</code>。设置为 <code>True</code> 时,所有非 <code>ASCII</code> 字符将被转义;<code>False</code> 则保留原始字符,仅当<code>format_json</code>为<code>True</code>时有效。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>False</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td rowspan="3"><code>save_to_json()</code></td>
|
|
|
|
|
<td rowspan="3">将结果保存为json格式的文件</td>
|
|
|
|
|
<td><code>save_path</code></td>
|
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>保存的文件路径,当为目录时,保存文件命名与输入文件类型命名一致。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>无</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>indent</code></td>
|
|
|
|
|
<td><code>int</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>指定缩进级别,以美化输出的 <code>JSON</code> 数据,使其更具可读性,仅当 <code>format_json</code> 为 <code>True</code> 时有效。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>4</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>ensure_ascii</code></td>
|
|
|
|
|
<td><code>bool</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>控制是否将非 <code>ASCII</code> 字符转义为 <code>Unicode</code>。设置为 <code>True</code> 时,所有非 <code>ASCII</code> 字符将被转义;<code>False</code> 则保留原始字符,仅当<code>format_json</code>为<code>True</code>时有效。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td><code>False</code></td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>save_to_img()</code></td>
|
|
|
|
|
<td>将结果保存为图像格式的文件</td>
|
|
|
|
|
<td><code>save_path</code></td>
|
|
|
|
|
<td><code>str</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>保存的文件路径,支持目录或文件路径。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>无</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</table>
|
|
|
|
|
|
|
|
|
|
- 调用`print()` 方法会将结果打印到终端,打印到终端的内容解释如下:
|
|
|
|
|
|
|
|
|
|
- `input_path`: `(str)` 待预测图像的输入路径
|
|
|
|
|
|
|
|
|
|
- `page_index`: `(Union[int, None])` 如果输入是PDF文件,则表示当前是PDF的第几页,否则为 `None`
|
|
|
|
|
|
|
|
|
|
- `model_settings`: `(Dict[str, bool])` 配置产线所需的模型参数
|
|
|
|
|
|
|
|
|
|
- `use_doc_preprocessor`: `(bool)` 控制是否启用文档预处理子产线
|
|
|
|
|
- `use_layout_detection`: `(bool)` 控制是否启用版面区域检测模块
|
|
|
|
|
|
|
|
|
|
- `doc_preprocessor_res`: `(Dict[str, Union[str, Dict[str, bool], int]])` 文档预处理子产线的输出结果。仅当`use_doc_preprocessor=True`时存在
|
|
|
|
|
- `input_path`: `(Union[str, None])` 图像预处理子产线接受的图像路径,当输入为`numpy.ndarray`时,保存为`None`
|
|
|
|
|
- `model_settings`: `(Dict)` 预处理子产线的模型配置参数
|
|
|
|
|
- `use_doc_orientation_classify`: `(bool)` 控制是否启用文档方向分类
|
|
|
|
|
- `use_doc_unwarping`: `(bool)` 控制是否启用文本图像矫正
|
|
|
|
|
- `angle`: `(int)` 文档方向分类的预测结果。启用时取值为[0,1,2,3],分别对应[0°,90°,180°,270°];未启用时为-1
|
|
|
|
|
- `layout_det_res`: `(Dict[str, List[Dict]])` 版面区域检测模块的输出结果。仅当`use_layout_detection=True`时存在
|
|
|
|
|
- `input_path`: `(Union[str, None])` 版面区域检测模块接收的图像路径,当输入为`numpy.ndarray`时,保存为`None`
|
|
|
|
|
- `boxes`: `(List[Dict[int, str, float, List[float]]])` 版面区域检测预测结果列表
|
|
|
|
|
- `cls_id`: `(int)` 版面区域检测预测的类别id
|
|
|
|
|
- `label`: `(str)` 版面区域检测预测的类别
|
|
|
|
|
- `score`: `(float)` 版面区域检测预测的类别置信度分数
|
|
|
|
|
- `coordinate`: `(List[float])` 版面区域检测预测的边界框坐标,格式为[x_min, y_min, x_max, y_max],其中(x_min, y_min)为左上角坐标,(x_max, y_max) 为右上角坐标
|
|
|
|
|
- `formula_res_list`: `(List[Dict[str, int, List[float]]])` 公式识别的预测结果列表
|
|
|
|
|
- `rec_formula`: `(str)` 公式识别预测的LaTeX源码
|
|
|
|
|
- `formula_region_id`: `(int)` 公式识别预测的id编号
|
|
|
|
|
- `dt_polys`: `(List[float])` 公式识别预测的边界框坐标,格式为[x_min, y_min, x_max, y_max],其中(x_min, y_min)为左上角坐标,(x_max, y_max) 为右上角坐标
|
|
|
|
|
|
|
|
|
|
- 调用`save_to_json()` 方法会将上述内容保存到指定的`save_path`中,如果指定为目录,则保存的路径为`save_path/{your_img_basename}_res.json`,如果指定为文件,则直接保存到该文件中。由于json文件不支持保存numpy数组,因此会将其中的`numpy.array`类型转换为列表形式。
|
|
|
|
|
- 调用`save_to_img()` 方法会将可视化结果保存到指定的`save_path`中,如果指定为目录,则保存的路径为`save_path/{your_img_basename}_formula_res_img.{your_img_extension}`,如果指定为文件,则直接保存到该文件中。(产线通常包含较多结果图片,不建议直接指定为具体的文件路径,否则多张图会被覆盖,仅保留最后一张图)
|
|
|
|
|
|
|
|
|
|
* 此外,也支持通过属性获取带结果的可视化图像和预测结果,具体如下:
|
|
|
|
|
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>属性</th>
|
|
|
|
|
<th>属性说明</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<td rowspan="1"><code>json</code></td>
|
|
|
|
|
<td rowspan="1">获取预测的 <code>json</code> 格式的结果</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td rowspan="2"><code>img</code></td>
|
|
|
|
|
<td rowspan="2">获取格式为 <code>dict</code> 的可视化图像</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</table>
|
|
|
|
|
|
|
|
|
|
- `json` 属性获取的预测结果为dict类型的数据,相关内容与调用 `save_to_json()` 方法保存的内容一致。
|
|
|
|
|
- `img` 属性返回的预测结果是一个字典类型的数据。其中,键分别为 `preprocessed_img`、 `layout_det_res`和 `formula_res_img`,对应的值是三个 `Image.Image` 对象:第一个用于展示图像预处理的可视化图像,第二个用于展示版面区域检测的可视化图像,第三个用于展示公式识别的可视化图像。如果没有使用图像预处理子模块,则字典中不包含 `preprocessed_img`;如果没有使用版面区域检测子模块,则字典中不包含`layout_det_res`。
|
|
|
|
|
|
|
|
|
|
## 3. 开发集成/部署
|
|
|
|
|
|
|
|
|
|
如果产线可以达到您对产线推理速度和精度的要求,您可以直接进行开发集成/部署。
|
|
|
|
|
|
|
|
|
|
若您需要将产线直接应用在您的Python项目中,可以参考 [2.2 Python脚本方式](#22-python脚本方式集成)中的示例代码。
|
|
|
|
|
|
|
|
|
|
此外,PaddleOCR 也提供了其他两种部署方式,详细说明如下:
|
|
|
|
|
|
2025-05-20 15:13:40 +08:00
|
|
|
|
🚀 高性能推理:在实际生产环境中,许多应用对部署策略的性能指标(尤其是响应速度)有着较严苛的标准,以确保系统的高效运行与用户体验的流畅性。为此,PaddleOCR 提供高性能推理功能,旨在对模型推理及前后处理进行深度性能优化,实现端到端流程的显著提速,详细的高性能推理流程请参考[高性能推理](../deployment/high_performance_inference.md)。
|
2025-05-18 21:09:53 +08:00
|
|
|
|
|
2025-05-20 15:13:40 +08:00
|
|
|
|
☁️ 服务化部署:服务化部署是实际生产环境中常见的一种部署形式。通过将推理功能封装为服务,客户端可以通过网络请求来访问这些服务,以获取推理结果。详细的产线服务化部署流程请参考[服务化部署](../deployment/serving.md)。
|
2025-05-18 21:09:53 +08:00
|
|
|
|
|
|
|
|
|
以下是基础服务化部署的API参考与多语言服务调用示例:
|
|
|
|
|
|
|
|
|
|
<details><summary>API参考</summary>
|
|
|
|
|
<p>对于服务提供的主要操作:</p>
|
|
|
|
|
<ul>
|
|
|
|
|
<li>HTTP请求方法为POST。</li>
|
|
|
|
|
<li>请求体和响应体均为JSON数据(JSON对象)。</li>
|
|
|
|
|
<li>当请求处理成功时,响应状态码为<code>200</code>,响应体的属性如下:</li>
|
|
|
|
|
</ul>
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>名称</th>
|
|
|
|
|
<th>类型</th>
|
|
|
|
|
<th>含义</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>logId</code></td>
|
|
|
|
|
<td><code>string</code></td>
|
|
|
|
|
<td>请求的UUID。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>errorCode</code></td>
|
|
|
|
|
<td><code>integer</code></td>
|
|
|
|
|
<td>错误码。固定为<code>0</code>。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>errorMsg</code></td>
|
|
|
|
|
<td><code>string</code></td>
|
|
|
|
|
<td>错误说明。固定为<code>"Success"</code>。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>result</code></td>
|
|
|
|
|
<td><code>object</code></td>
|
|
|
|
|
<td>操作结果。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
<ul>
|
|
|
|
|
<li>当请求处理未成功时,响应体的属性如下:</li>
|
|
|
|
|
</ul>
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>名称</th>
|
|
|
|
|
<th>类型</th>
|
|
|
|
|
<th>含义</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>logId</code></td>
|
|
|
|
|
<td><code>string</code></td>
|
|
|
|
|
<td>请求的UUID。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>errorCode</code></td>
|
|
|
|
|
<td><code>integer</code></td>
|
|
|
|
|
<td>错误码。与响应状态码相同。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>errorMsg</code></td>
|
|
|
|
|
<td><code>string</code></td>
|
|
|
|
|
<td>错误说明。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
<p>服务提供的主要操作如下:</p>
|
|
|
|
|
<ul>
|
|
|
|
|
<li><b><code>infer</code></b></li>
|
|
|
|
|
</ul>
|
|
|
|
|
<p>获取图像公式识别结果。</p>
|
|
|
|
|
<p><code>POST /formula-recognition</code></p>
|
|
|
|
|
<ul>
|
|
|
|
|
<li>请求体的属性如下:</li>
|
|
|
|
|
</ul>
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>名称</th>
|
|
|
|
|
<th>类型</th>
|
|
|
|
|
<th>含义</th>
|
|
|
|
|
<th>是否必填</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>file</code></td>
|
|
|
|
|
<td><code>string</code></td>
|
|
|
|
|
<td>服务器可访问的图像文件或PDF文件的URL,或上述类型文件内容的Base64编码结果。默认对于超过10页的PDF文件,只有前10页的内容会被处理。<br /> 要解除页数限制,请在产线配置文件中添加以下配置:
|
|
|
|
|
<pre><code>Serving:
|
|
|
|
|
extra:
|
|
|
|
|
max_num_input_imgs: null
|
|
|
|
|
</code></pre>
|
|
|
|
|
</td>
|
|
|
|
|
<td>是</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>fileType</code></td>
|
|
|
|
|
<td><code>integer</code> | <code>null</code></td>
|
|
|
|
|
<td>文件类型。<code>0</code>表示PDF文件,<code>1</code>表示图像文件。若请求体无此属性,则将根据URL推断文件类型。</td>
|
|
|
|
|
<td>否</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>useDocOrientationClassify</code></td>
|
|
|
|
|
<td><code>boolean</code> | <code>null</code></td>
|
|
|
|
|
<td>请参阅产线对象中 <code>predict</code> 方法的 <code>use_doc_orientation_classify</code> 参数相关说明。</td>
|
|
|
|
|
<td>否</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>useDocUnwarping</code></td>
|
|
|
|
|
<td><code>boolean</code> | <code>null</code></td>
|
|
|
|
|
<td>请参阅产线对象中 <code>predict</code> 方法的 <code>use_doc_unwarping</code> 参数相关说明。</td>
|
|
|
|
|
<td>否</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>useLayoutDetection</code></td>
|
|
|
|
|
<td><code>boolean</code> | <code>null</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>请参阅产线对象中 <code>predict</code>方法的 <code>use_layout_detection</code> 参数相关说明。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>否</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layoutThreshold</code></td>
|
|
|
|
|
<td><code>number</code> | <code>null</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>请参阅产线对象中<code>predict</code> 法的<code>layout_threshold</code> 参数相关说明。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>否</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layoutNms</code></td>
|
|
|
|
|
<td><code>boolean</code> | <code>null</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>请参阅产线对象中<code>predict</code> 法的<code>layout_nms</code> 参数相关说明。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>否</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layoutUnclipRatio</code></td>
|
|
|
|
|
<td><code>number</code> | <code>array</code> | <code>null</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>请参阅产线对象中 <code>predict</code>方法的 <code>layout_unclip_ratio</code> 参数相关说明。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>否</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>layoutMergeBboxesMode</code></td>
|
|
|
|
|
<td><code>string</code> | <code>null</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>请参阅产线对象中<code>predict</code>方法的 <code>layout_merge_bboxes_mode</code> 参数相关说明。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
<td>否</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
<ul>
|
|
|
|
|
<li>请求处理成功时,响应体的<code>result</code>具有如下属性:</li>
|
|
|
|
|
</ul>
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>名称</th>
|
|
|
|
|
<th>类型</th>
|
|
|
|
|
<th>含义</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>formulaRecResults</code></td>
|
|
|
|
|
<td><code>object</code></td>
|
|
|
|
|
<td>公式识别结果。数组长度为1(对于图像输入)或实际处理的文档页数(对于PDF输入)。对于PDF输入,数组中的每个元素依次表示PDF文件中实际处理的每一页的结果。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>dataInfo</code></td>
|
|
|
|
|
<td><code>object</code></td>
|
|
|
|
|
<td>输入数据信息。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
<p><code>formulaRecResults</code>中的每个元素为一个<code>object</code>,具有如下属性:</p>
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>名称</th>
|
|
|
|
|
<th>类型</th>
|
|
|
|
|
<th>含义</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>prunedResult</code></td>
|
|
|
|
|
<td><code>object</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>产线对象的 <code>predict</code> 方法生成结果的JSON表示中<code>res</code>字段的简化版本,其中去除了<code>input_path</code>和 <code>page_index</code>字段。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>outputImages</code></td>
|
|
|
|
|
<td><code>object</code> | <code>null</code></td>
|
2025-05-28 10:24:37 +08:00
|
|
|
|
<td>参见产线预测结果的<code>img</code>属性说明。图像为JPEG格式,使用Base64编码。</td>
|
2025-05-18 21:09:53 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td><code>inputImage</code> | <code>null</code></td>
|
|
|
|
|
<td><code>string</code></td>
|
|
|
|
|
<td>输入图像。图像为JPEG格式,使用Base64编码。</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
|
|
|
|
</details>
|
|
|
|
|
<details><summary>多语言调用服务示例</summary>
|
|
|
|
|
<details>
|
|
|
|
|
<summary>Python</summary>
|
|
|
|
|
|
|
|
|
|
<pre><code class="language-python">import base64
|
|
|
|
|
import requests
|
|
|
|
|
|
|
|
|
|
API_URL = "http://localhost:8080/formula-recognition"
|
|
|
|
|
file_path = "./demo.jpg"
|
|
|
|
|
|
|
|
|
|
with open(file_path, "rb") as file:
|
|
|
|
|
file_bytes = file.read()
|
|
|
|
|
file_data = base64.b64encode(file_bytes).decode("ascii")
|
|
|
|
|
|
|
|
|
|
payload = {"file": file_data, "fileType": 1}
|
|
|
|
|
|
|
|
|
|
response = requests.post(API_URL, json=payload)
|
|
|
|
|
|
|
|
|
|
assert response.status_code == 200
|
|
|
|
|
result = response.json()["result"]
|
|
|
|
|
for i, res in enumerate(result["formulaRecResults"]):
|
|
|
|
|
print(res["prunedResult"])
|
|
|
|
|
for img_name, img in res["outputImages"].items():
|
|
|
|
|
img_path = f"{img_name}_{i}.jpg"
|
|
|
|
|
with open(img_path, "wb") as f:
|
|
|
|
|
f.write(base64.b64decode(img))
|
|
|
|
|
print(f"Output image saved at {img_path}")
|
|
|
|
|
</code></pre></details>
|
|
|
|
|
</details>
|
|
|
|
|
<br/>
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
## 4. 二次开发
|
|
|
|
|
|
|
|
|
|
如果公式识别产线提供的默认模型权重在您的场景中,精度或速度不满意,您可以尝试利用<b>您自己拥有的特定领域或应用场景的数据</b>对现有模型进行进一步的<b>微调</b>,以提升公式识别产线的在您的场景中的识别效果。
|
|
|
|
|
|
2025-05-27 20:14:53 +08:00
|
|
|
|
### 4.1 模型微调
|
2025-05-20 00:28:19 +08:00
|
|
|
|
由于公式识别产线包含若干模块,模型产线的效果如果不及预期,可能来自于其中任何一个模块。您可以对识别效果差的图片进行分析,进而确定是哪个模块存在问题,并参考以下表格中对应的微调教程链接进行模型微调。
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
<table>
|
|
|
|
|
<thead>
|
|
|
|
|
<tr>
|
|
|
|
|
<th>情形</th>
|
|
|
|
|
<th>微调模块</th>
|
|
|
|
|
<th>微调参考链接</th>
|
|
|
|
|
</tr>
|
|
|
|
|
</thead>
|
|
|
|
|
<tbody>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>公式存在漏检</td>
|
|
|
|
|
<td>版面区域检测模块</td>
|
2025-05-27 20:14:53 +08:00
|
|
|
|
<td><a href="https://paddlepaddle.github.io/PaddleOCR/latest/version3.x/module_usage/layout_detection.html#_5">链接</a></td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>公式内容不准</td>
|
|
|
|
|
<td>公式识别模块</td>
|
2025-05-27 20:14:53 +08:00
|
|
|
|
<td><a href="https://paddlepaddle.github.io/PaddleOCR/latest/version3.x/module_usage/formula_recognition.html#_5">链接</a></td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>整图旋转矫正不准</td>
|
|
|
|
|
<td>文档图像方向分类模块</td>
|
2025-05-27 20:14:53 +08:00
|
|
|
|
<td><a href="https://paddlepaddle.github.io/PaddleOCR/latest/version3.x/module_usage/doc_img_orientation_classification.html#_5">链接</a></td>
|
2025-05-20 00:28:19 +08:00
|
|
|
|
</tr>
|
|
|
|
|
<tr>
|
|
|
|
|
<td>图像扭曲矫正不准</td>
|
|
|
|
|
<td>文本图像矫正模块</td>
|
|
|
|
|
<td>暂不支持微调</td>
|
|
|
|
|
</tr>
|
|
|
|
|
</tbody>
|
|
|
|
|
</table>
|
2025-05-27 20:14:53 +08:00
|
|
|
|
|
|
|
|
|
### 4.2 模型应用
|
|
|
|
|
|
|
|
|
|
当您使用私有数据集完成微调训练后,可获得本地模型权重文件,然后可以通过参数指定本地模型保存路径的方式,或者通过自定义产线配置文件的方式,使用微调后的模型权重。
|
|
|
|
|
|
|
|
|
|
#### 4.2.1 通过参数指定本地模型路径
|
|
|
|
|
|
|
|
|
|
在初始化产线对象时,通过参数指定本地模型路径。以公式识别模型微调后的权重的使用方法为例,示例如下:
|
|
|
|
|
|
|
|
|
|
命令行方式:
|
|
|
|
|
|
|
|
|
|
```bash
|
|
|
|
|
# 通过 --formula_recognition_model_dir 指定本地模型路径
|
|
|
|
|
paddleocr formula_recognition_pipeline -i ./general_formula_recognition_001.png --formula_recognition_model_dir your_formula_recognition_model_path
|
|
|
|
|
|
|
|
|
|
# 默认使用 PP-FormulaNet_plus-M 模型作为默认公式识别模型,如果微调的不是该模型,通过 --formula_recognition_model_name 修改模型名称
|
|
|
|
|
paddleocr formula_recognition_pipeline -i ./general_formula_recognition_001.png --formula_recognition_model_name PP-FormulaNet_plus-M --formula_recognition_model_dir your_ppformulanet_plus-m_formula_recognition_model_path
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
脚本方式:
|
|
|
|
|
|
|
|
|
|
```python
|
|
|
|
|
|
|
|
|
|
from paddleocr import FormulaRecognitionPipeline
|
|
|
|
|
|
|
|
|
|
# 通过 formula_recognition_model_dir 指定本地模型路径
|
|
|
|
|
pipeline = FormulaRecognitionPipeline(formula_recognition_model_dir="./your_formula_recognition_model_path")
|
|
|
|
|
output = pipeline.predict("./general_formula_recognition_001.png")
|
|
|
|
|
for res in output:
|
|
|
|
|
res.print() ## 打印预测的结构化输出
|
|
|
|
|
res.save_to_img(save_path="output") ## 保存当前图像的公式可视化结果
|
|
|
|
|
res.save_to_json(save_path="output") ## 保存当前图像的结构化json结果
|
|
|
|
|
# 默认使用 PP-FormulaNet_plus-M 模型作为默认公式识别模型,如果微调的不是该模型,通过 formula_recognition_model_name 修改模型名称
|
|
|
|
|
# pipeline = FormulaRecognitionPipeline(formula_recognition_model_name="PP-FormulaNet_plus-M", formula_recognition_model_dir="./your_ppformulanet_plus-m_formula_recognition_model_path")
|
|
|
|
|
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
#### 4.2.2 通过配置文件指定本地模型路径
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1.获取产线配置文件
|
|
|
|
|
|
|
|
|
|
可调用 PaddleOCR 中 公式识别 产线对象的 `export_paddlex_config_to_yaml` 方法,将当前产线配置导出为 YAML 文件:
|
|
|
|
|
|
|
|
|
|
```Python
|
|
|
|
|
from paddleocr import FormulaRecognitionPipeline
|
|
|
|
|
|
|
|
|
|
pipeline = FormulaRecognitionPipeline()
|
|
|
|
|
pipeline.export_paddlex_config_to_yaml("FormulaRecognitionPipeline.yaml")
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
2.修改配置文件
|
|
|
|
|
|
|
|
|
|
在得到默认的产线配置文件后,将微调后模型权重的本地路径替换至产线配置文件中的对应位置即可。例如
|
|
|
|
|
|
|
|
|
|
```yaml
|
|
|
|
|
......
|
|
|
|
|
SubModules:
|
|
|
|
|
FormulaRecognition:
|
|
|
|
|
batch_size: 5
|
|
|
|
|
model_dir: null # 替换为微调后的公式模型权重路径
|
|
|
|
|
model_name: PP-FormulaNet_plus-M # 如果微调的模型名称与默认模型名称不同,请一并修改此处
|
|
|
|
|
module_name: formula_recognition
|
|
|
|
|
LayoutDetection:
|
|
|
|
|
batch_size: 1
|
|
|
|
|
layout_merge_bboxes_mode: large
|
|
|
|
|
layout_nms: true
|
|
|
|
|
layout_unclip_ratio: 1.0
|
|
|
|
|
model_dir: null # 替换为微调后的版面区域检测模型权重路径
|
|
|
|
|
model_name: PP-DocLayout_plus-L # 如果微调的模型名称与默认模型名称不同,请一并修改此处
|
|
|
|
|
module_name: layout_detection
|
|
|
|
|
threshold: 0.5
|
|
|
|
|
SubPipelines:
|
|
|
|
|
DocPreprocessor:
|
|
|
|
|
SubModules:
|
|
|
|
|
DocOrientationClassify:
|
|
|
|
|
batch_size: 1
|
|
|
|
|
model_dir: null # 替换为微调后的文档图像方向分类模型权重路径
|
|
|
|
|
model_name: PP-LCNet_x1_0_doc_ori # 如果微调的模型名称与默认模型名称不同,请一并修改此处
|
|
|
|
|
module_name: doc_text_orientation
|
|
|
|
|
DocUnwarping:
|
|
|
|
|
batch_size: 1
|
|
|
|
|
model_dir: null
|
|
|
|
|
model_name: UVDoc
|
|
|
|
|
module_name: image_unwarping
|
|
|
|
|
pipeline_name: doc_preprocessor
|
|
|
|
|
use_doc_orientation_classify: true
|
|
|
|
|
use_doc_unwarping: true
|
|
|
|
|
pipeline_name: formula_recognition
|
|
|
|
|
use_doc_preprocessor: true
|
|
|
|
|
use_layout_detection: true
|
|
|
|
|
......
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
在产线配置文件中,不仅包含 PaddleOCR CLI 和 Python API 支持的参数,还可进行更多高级配置,具体信息可在 [PaddleX模型产线使用概览](https://paddlepaddle.github.io/PaddleX/3.0/pipeline_usage/pipeline_develop_guide.html) 中找到对应的产线使用教程,参考其中的详细说明,根据需求调整各项配置。
|
|
|
|
|
|
|
|
|
|
3.在 CLI 中加载产线配置文件
|
|
|
|
|
|
|
|
|
|
在修改完成配置文件后,通过命令行的 --paddlex_config 参数指定修改后的产线配置文件的路径,PaddleOCR 会读取其中的内容作为产线配置。示例如下:
|
|
|
|
|
|
|
|
|
|
```bash
|
|
|
|
|
paddleocr formula_recognition_pipeline -i ./general_formula_recognition_001.png --paddlex_config FormulaRecognitionPipeline.yaml
|
|
|
|
|
```
|
|
|
|
|
|
|
|
|
|
4.在 Python API 中加载产线配置文件
|
|
|
|
|
|
|
|
|
|
初始化产线对象时,可通过 paddlex_config 参数传入 PaddleX 产线配置文件路径或配置字典,PaddleOCR 会读取其中的内容作为产线配置。示例如下:
|
|
|
|
|
|
|
|
|
|
```python
|
|
|
|
|
from paddleocr import FormulaRecognitionPipeline
|
|
|
|
|
|
|
|
|
|
pipeline = FormulaRecognitionPipeline(paddlex_config="FormulaRecognitionPipeline.yaml")
|
|
|
|
|
output = pipeline.predict("./general_formula_recognition_001.png")
|
|
|
|
|
for res in output:
|
|
|
|
|
res.print() ## 打印预测的结构化输出
|
|
|
|
|
res.save_to_img(save_path="output") ## 保存当前图像的公式可视化结果
|
|
|
|
|
res.save_to_json(save_path="output") ## 保存当前图像的结构化json结果
|
|
|
|
|
```
|