quickstart.md 6.1 KB
Newer Older
D
dyning 已提交
1

D
dyning 已提交
2
# 中文OCR模型快速使用
D
dyning 已提交
3 4 5 6 7

## 1.环境配置

请先参考[快速安装](./installation.md)配置PaddleOCR运行环境。

D
dyning 已提交
8
*注意:也可以通过 whl 包安装使用PaddleOCR,具体参考[Paddleocr Package使用说明](https://github.com/PaddlePaddle/PaddleOCR/blob/develop/doc/doc_ch/whl.md)。*
D
dyning 已提交
9

D
dyning 已提交
10 11
## 2.inference模型下载

littletomatodonkey's avatar
littletomatodonkey 已提交
12 13 14 15 16
* 移动端和服务器端的检测与识别模型如下,更多模型下载(包括多语言),可以参考[PP-OCR v1.1 系列模型下载](./doc_ch/models_list.md)
| 模型简介     | 模型名称     |推荐场景          | 检测模型 | 方向分类器 | 识别模型 |      |
| ------------ | --------------- | ----------------|---- | ---------- | -------- | ---- |
| 中英文超轻量OCR模型(8.1M) | ch_ppocr_mobile_v1.1_xx |移动端&服务器端|[推理模型](https://paddleocr.bj.bcebos.com/20-09-22/mobile/det/ch_ppocr_mobile_v1.1_det_infer.tar) / [预训练模型](https://paddleocr.bj.bcebos.com/20-09-22/mobile/det/ch_ppocr_mobile_v1.1_det_train.tar)|[推理模型](https://paddleocr.bj.bcebos.com/20-09-22/cls/ch_ppocr_mobile-v1.1.cls_infer.tar) / [预训练模型](https://paddleocr.bj.bcebos.com/20-09-22/cls/ch_ppocr_mobile-v1.1.cls_train.tar) |[推理模型](https://paddleocr.bj.bcebos.com/20-09-22/mobile/rec/ch_ppocr_mobile_v1.1_rec_infer.tar) / [预训练模型](https://paddleocr.bj.bcebos.com/20-09-22/mobile/rec/ch_ppocr_mobile_v1.1_rec_pre.tar)      | |
| 中英文通用OCR模型(155.1M)   |ch_ppocr_server_v1.1_xx|服务器端 |[推理模型](https://paddleocr.bj.bcebos.com/20-09-22/server/det/ch_ppocr_server_v1.1_det_infer.tar) / [预训练模型](https://paddleocr.bj.bcebos.com/20-09-22/server/det/ch_ppocr_server_v1.1_det_train.tar)          |[推理模型](https://paddleocr.bj.bcebos.com/20-09-22/cls/ch_ppocr_mobile-v1.1.cls_infer.tar) / [预训练模型](https://paddleocr.bj.bcebos.com/20-09-22/cls/ch_ppocr_mobile-v1.1.cls_train.tar)    |[推理模型](https://paddleocr.bj.bcebos.com/20-09-22/server/rec/ch_ppocr_server_v1.1_rec_infer.tar) / [预训练模型](https://paddleocr.bj.bcebos.com/20-09-22/server/rec/ch_ppocr_server_v1.1_rec_pre.tar)  |  |
D
dyning 已提交
17

littletomatodonkey's avatar
littletomatodonkey 已提交
18 19

* windows 环境下如果没有安装wget,下载模型时可将链接复制到浏览器中下载,并解压放置在相应目录下*
D
dyning 已提交
20 21 22 23 24 25 26 27 28

复制上表中的检测和识别的`inference模型`下载地址,并解压

```
mkdir inference && cd inference
# 下载检测模型并解压
wget {url/of/detection/inference_model} && tar xf {name/of/detection/inference_model/package}
# 下载识别模型并解压
wget {url/of/recognition/inference_model} && tar xf {name/of/recognition/inference_model/package}
littletomatodonkey's avatar
littletomatodonkey 已提交
29 30
# 下载方向分类器模型并解压
wget {url/of/classification/inference_model} && tar xf {name/of/classification/inference_model/package}
D
dyning 已提交
31 32 33 34 35 36 37 38
cd ..
```

以超轻量级模型为例:

```
mkdir inference && cd inference
# 下载超轻量级中文OCR模型的检测模型并解压
littletomatodonkey's avatar
littletomatodonkey 已提交
39
wget https://paddleocr.bj.bcebos.com/20-09-22/mobile/det/ch_ppocr_mobile_v1.1_det_infer.tar && tar xf ch_ppocr_mobile_v1.1_det_infer.tar
D
dyning 已提交
40
# 下载超轻量级中文OCR模型的识别模型并解压
littletomatodonkey's avatar
littletomatodonkey 已提交
41 42 43
wget https://paddleocr.bj.bcebos.com/20-09-22/mobile/rec/ch_ppocr_mobile_v1.1_rec_infer.tar
# 下载超轻量级中文OCR模型的文本方向分类器模型并解压
wget https://paddleocr.bj.bcebos.com/20-09-22/cls/ch_ppocr_mobile-v1.1.cls_infer.tar
D
dyning 已提交
44 45 46 47 48 49 50
cd ..
```

解压完毕后应有如下文件结构:

```
|-inference
littletomatodonkey's avatar
littletomatodonkey 已提交
51 52 53 54
    |-ch_ppocr_mobile_v1.1_det_infer
        |- model
        |- params
    |-ch_ppocr_mobile_v1.1_rec_infer
D
dyning 已提交
55 56
        |- model
        |- params
littletomatodonkey's avatar
littletomatodonkey 已提交
57
    |-ch_ppocr_mobile-v1.1.cls_infer
D
dyning 已提交
58 59 60 61 62 63 64
        |- model
        |- params
    ...
```

## 3.单张图像或者图像集合预测

littletomatodonkey's avatar
littletomatodonkey 已提交
65
以下代码实现了文本检测、识别串联推理,在执行预测时,需要通过参数image_dir指定单张图像或者图像集合的路径、参数det_model_dir指定检测inference模型的路径、参数rec_model_dir指定识别inference模型的路径、参数use_angle_cls指定是否使用方向分类器,参数cls_model_dir指定方向分类器inference模型的路径。可视化识别结果默认保存到 ./inference_results 文件夹里面。
D
dyning 已提交
66 67 68 69

```bash

# 预测image_dir指定的单张图像
littletomatodonkey's avatar
littletomatodonkey 已提交
70
python3 tools/infer/predict_system.py --image_dir="./doc/imgs/11.jpg" --det_model_dir="./inference/ch_ppocr_mobile_v1.1_det_infer/"  --rec_model_dir="./inference/ch_ppocr_mobile_v1.1_rec_infer/" --cls_model_dir="./inference/ch_ppocr_mobile_v1.1_cls_infer/" --use_angle_cls=True --use_space_char=True
D
dyning 已提交
71 72

# 预测image_dir指定的图像集合
littletomatodonkey's avatar
littletomatodonkey 已提交
73
python3 tools/infer/predict_system.py --image_dir="./doc/imgs/" --det_model_dir="./inference/ch_ppocr_mobile_v1.1_det_infer/"  --rec_model_dir="./inference/ch_ppocr_mobile_v1.1_rec_infer/" --cls_model_dir="./inference/ch_ppocr_mobile_v1.1_cls_infer/" --use_angle_cls=True --use_space_char=True
D
dyning 已提交
74 75

# 如果想使用CPU进行预测,需设置use_gpu参数为False
littletomatodonkey's avatar
littletomatodonkey 已提交
76 77

python3 tools/infer/predict_system.py --image_dir="./doc/imgs/11.jpg" --det_model_dir="./inference/ch_ppocr_mobile_v1.1_det_infer/"  --rec_model_dir="./inference/ch_ppocr_mobile_v1.1_rec_infer/" --cls_model_dir="./inference/ch_ppocr_mobile_v1.1_cls_infer/" --use_angle_cls=True --use_space_char=True --use_gpu=False
D
dyning 已提交
78 79
```

T
tink2123 已提交
80 81 82
- 通用中文OCR模型

请按照上述步骤下载相应的模型,并且更新相关的参数,示例如下:
littletomatodonkey's avatar
littletomatodonkey 已提交
83
```bash
D
dyning 已提交
84
# 预测image_dir指定的单张图像
littletomatodonkey's avatar
littletomatodonkey 已提交
85
python3 tools/infer/predict_system.py --image_dir="./doc/imgs/11.jpg" --det_model_dir="./inference/ch_ppocr_server_v1.1_det_infer/"  --rec_model_dir="./inference/ch_ppocr_server_v1.1_rec_infer/" --cls_model_dir="./inference/ch_ppocr_mobile_v1.1_cls_infer/" --use_angle_cls=True --use_space_char=True
D
dyning 已提交
86 87
```

littletomatodonkey's avatar
littletomatodonkey 已提交
88
* **如果希望使用不支持空格的识别模型,在预测的时候需要注意:请将代码更新到最新版本,并添加参数 `--use_space_char=False`**
D
dyning 已提交
89

D
dyning 已提交
90
更多的文本检测、识别串联推理使用方式请参考文档教程中[基于Python预测引擎推理](./inference.md)
D
dyning 已提交
91 92

此外,文档教程中也提供了中文OCR模型的其他预测部署方式:
D
dyning 已提交
93
- [基于C++预测引擎推理](../../deploy/cpp_infer/readme.md)
littletomatodonkey's avatar
littletomatodonkey 已提交
94
- [服务部署](../../deploy/pdserving/readme.md)
D
dyning 已提交
95
- [端侧部署](../../deploy/lite/readme.md)