quickstart.md 6.53 KB
Newer Older
MissPenguin's avatar
update  
MissPenguin committed
1
# PaddleOCR 快速开始
Leif's avatar
Leif committed
2

MissPenguin's avatar
update  
MissPenguin committed
3
**说明:** 本文主要介绍PaddleOCR wheel包对PP-OCR系列模型的快速使用,如要体验文档分析相关功能,请参考[PP-Structure快速使用教程](../../ppstructure/docs/quickstart.md)
MissPenguin's avatar
update  
MissPenguin committed
4

Leif's avatar
Leif committed
5
6
7
- [1. 安装](#1)
  - [1.1 安装PaddlePaddle](#11)
  - [1.2 安装PaddleOCR whl包](#12)
MissPenguin's avatar
MissPenguin committed
8
9
- [2. 便捷使用](#2)
  - [2.1 命令行使用](#21)
Evezerest's avatar
Evezerest committed
10
11
      - [2.1.1 中英文模型](#211)
      - [2.1.2 多语言模型](#212)
MissPenguin's avatar
MissPenguin committed
12
  - [2.2 Python脚本使用](#22)
Evezerest's avatar
Evezerest committed
13
      - [2.2.1 中英文与多语言使用](#221)
Leif's avatar
Leif committed
14
- [3.小结](#3)
Leif's avatar
Leif committed
15

WenmuZhou's avatar
WenmuZhou committed
16

Evezerest's avatar
Evezerest committed
17
<a name="1"></a>
Leif's avatar
Leif committed
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
## 1. 安装

<a name="11"></a>
### 1.1 安装PaddlePaddle

> 如果您没有基础的Python运行环境,请参考[运行环境准备](./environment.md)。

- 您的机器安装的是CUDA9或CUDA10,请运行以下命令安装

  ```bash
  python3 -m pip install paddlepaddle-gpu -i https://mirror.baidu.com/pypi/simple
  ```

- 您的机器是CPU,请运行以下命令安装

  ```bash
  python3 -m pip install paddlepaddle -i https://mirror.baidu.com/pypi/simple
  ```

更多的版本需求,请参照[飞桨官网安装文档](https://www.paddlepaddle.org.cn/install/quick)中的说明进行操作。

<a name="12"></a>
### 1.2 安装PaddleOCR whl包
Leif's avatar
Leif committed
41
42
43

```bash
pip install "paddleocr>=2.0.1" # 推荐使用2.0.1+版本
dyning's avatar
dyning committed
44
45
```

MissPenguin's avatar
update  
MissPenguin committed
46
- 对于Windows环境用户:直接通过pip安装的shapely库可能出现`[winRrror 126] 找不到指定模块的问题`。建议从[这里](https://www.lfd.uci.edu/~gohlke/pythonlibs/#shapely)下载shapely安装包完成安装。
Leif's avatar
Leif committed
47
48


Leif's avatar
Leif committed
49
<a name="2"></a>
Leif's avatar
Leif committed
50
## 2. 便捷使用
Leif's avatar
Leif committed
51
<a name="21"></a>
Leif's avatar
Leif committed
52
53
### 2.1 命令行使用

Leif's avatar
Leif committed
54
PaddleOCR提供了一系列测试图片,点击[这里](https://paddleocr.bj.bcebos.com/dygraph_v2.1/ppocr_img.zip)下载并解压,然后在终端中切换到相应目录
dyning's avatar
dyning committed
55
56

```
Leif's avatar
Leif committed
57
cd /path/to/ppocr_img
dyning's avatar
dyning committed
58
59
```

MissPenguin's avatar
update  
MissPenguin committed
60
61
如果不使用提供的测试图片,可以将下方`--image_dir`参数替换为相应的测试图片路径。

62
63
**注意** whl包默认使用`PP-OCRv3`模型,识别模型使用的输入shape为`3,48,320`, 因此如果使用识别功能,需要添加参数`--rec_image_shape 3,48,320`,如果不使用默认的`PP-OCRv3`模型,则无需设置该参数。

Leif's avatar
Leif committed
64
<a name="211"></a>
Leif's avatar
Leif committed
65
66
#### 2.1.1 中英文模型

67
* 检测+方向分类器+识别全流程:`--use_angle_cls true`设置使用方向分类器识别180度旋转文字,`--use_gpu false`设置不使用GPU
Leif's avatar
Leif committed
68
69

  ```bash
70
  paddleocr --image_dir ./imgs/11.jpg --use_angle_cls true --use_gpu false --rec_image_shape 3,48,320
Leif's avatar
Leif committed
71
72
73
74
75
  ```

  结果是一个list,每个item包含了文本框,文字和识别置信度

  ```bash
76
  [[[28.0, 37.0], [302.0, 39.0], [302.0, 72.0], [27.0, 70.0]], ('纯臻营养护发素', 0.9658738374710083)]
Leif's avatar
Leif committed
77
78
79
80
81
82
83
84
85
86
87
88
  ......
  ```

- 单独使用检测:设置`--rec``false`

  ```bash
  paddleocr --image_dir ./imgs/11.jpg --rec false
  ```

  结果是一个list,每个item只包含文本框

  ```bash
89
90
  [[27.0, 459.0], [136.0, 459.0], [136.0, 479.0], [27.0, 479.0]]
  [[28.0, 429.0], [372.0, 429.0], [372.0, 445.0], [28.0, 445.0]]
Leif's avatar
Leif committed
91
92
93
94
95
96
  ......
  ```

- 单独使用识别:设置`--det``false`

  ```bash
97
  paddleocr --image_dir ./imgs_words/ch/word_1.jpg --det false --rec_image_shape 3,48,320
Leif's avatar
Leif committed
98
99
100
101
102
  ```

  结果是一个list,每个item只包含识别结果和识别置信度

  ```bash
103
  ['韩国小馆', 0.994467]
Leif's avatar
Leif committed
104
105
106
  ```


107
如需使用2.0模型,请指定参数`--version PP-OCR`,paddleocr默认使用PP-OCRv3模型(`--versioin PP-OCRv3`)。更多whl包使用可参考[whl包文档](./whl.md)
Leif's avatar
Leif committed
108

Leif's avatar
Leif committed
109
<a name="212"></a>
110

Leif's avatar
Leif committed
111
112
#### 2.1.2 多语言模型

113
Paddleocr目前支持80个语种,可以通过修改`--lang`参数进行切换,对于英文模型,指定`--lang=en`, PP-OCRv3目前只支持中文和英文模型,其他多语言模型会陆续更新。
Leif's avatar
Leif committed
114
115

``` bash
116
paddleocr --image_dir ./imgs_en/254.jpg --lang=en --rec_image_shape 3,48,320
dyning's avatar
dyning committed
117
```
Leif's avatar
Leif committed
118
119
120
121
122
123
124
125
126

<div align="center">
    <img src="../imgs_en/254.jpg" width="300" height="600">
    <img src="../imgs_results/multi_lang/img_02.jpg" width="600" height="600">
</div>

结果是一个list,每个item包含了文本框,文字和识别置信度

```text
127
128
129
[[[67.0, 51.0], [327.0, 46.0], [327.0, 74.0], [68.0, 80.0]], ('PHOCAPITAL', 0.9944712519645691)]
[[[72.0, 92.0], [453.0, 84.0], [454.0, 114.0], [73.0, 122.0]], ('107 State Street', 0.9744491577148438)]
[[[69.0, 135.0], [501.0, 125.0], [501.0, 156.0], [70.0, 165.0]], ('Montpelier Vermont', 0.9357033967971802)]
Leif's avatar
Leif committed
130
......
dyning's avatar
dyning committed
131
132
```

Leif's avatar
Leif committed
133
常用的多语言简写包括
dyning's avatar
dyning committed
134

Leif's avatar
Leif committed
135
136
137
138
139
| 语种     | 缩写        |      | 语种     | 缩写   |      | 语种     | 缩写   |
| -------- | ----------- | ---- | -------- | ------ | ---- | -------- | ------ |
| 中文     | ch          |      | 法文     | fr     |      | 日文     | japan  |
| 英文     | en          |      | 德文     | german |      | 韩文     | korean |
| 繁体中文 | chinese_cht |      | 意大利文 | it     |      | 俄罗斯文 | ru     |
dyning's avatar
dyning committed
140

Leif's avatar
Leif committed
141
全部语种及其对应的缩写列表可查看[多语言模型教程](./multi_languages.md)
142

Leif's avatar
Leif committed
143

Leif's avatar
Leif committed
144
<a name="22"></a>
Leif's avatar
Leif committed
145
### 2.2 Python脚本使用
Leif's avatar
Leif committed
146
<a name="221"></a>
Leif's avatar
Leif committed
147
148
#### 2.2.1 中英文与多语言使用

Leif's avatar
Leif committed
149
通过Python脚本使用PaddleOCR whl包,whl包会自动下载ppocr轻量级模型作为默认模型。
Leif's avatar
Leif committed
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176

* 检测+方向分类器+识别全流程

```python
from paddleocr import PaddleOCR, draw_ocr

# Paddleocr目前支持的多语言语种可以通过修改lang参数进行切换
# 例如`ch`, `en`, `fr`, `german`, `korean`, `japan`
ocr = PaddleOCR(use_angle_cls=True, lang="ch")  # need to run only once to download and load model into memory
img_path = './imgs/11.jpg'
result = ocr.ocr(img_path, cls=True)
for line in result:
    print(line)

# 显示结果
from PIL import Image

image = Image.open(img_path).convert('RGB')
boxes = [line[0] for line in result]
txts = [line[1][0] for line in result]
scores = [line[1][1] for line in result]
im_show = draw_ocr(image, boxes, txts, scores, font_path='./fonts/simfang.ttf')
im_show = Image.fromarray(im_show)
im_show.save('result.jpg')
```

结果是一个list,每个item包含了文本框,文字和识别置信度
WenmuZhou's avatar
WenmuZhou committed
177
178

```bash
179
[[[28.0, 37.0], [302.0, 39.0], [302.0, 72.0], [27.0, 70.0]], ('纯臻营养护发素', 0.9658738374710083)]
Leif's avatar
Leif committed
180
......
dyning's avatar
dyning committed
181
182
```

Leif's avatar
Leif committed
183
184
185
186
187
结果可视化

<div align="center">
    <img src="../imgs_results/whl/11_det_rec.jpg" width="800">
</div>
188

Leif's avatar
Leif committed
189
190
191
192
193
194
195

<a name="3"></a>

## 3. 小结

通过本节内容,相信您已经熟练掌握PaddleOCR whl包的使用方法并获得了初步效果。

MissPenguin's avatar
update  
MissPenguin committed
196
PaddleOCR是一套丰富领先实用的OCR工具库,打通数据、模型训练、压缩和推理部署全流程,您可以参考[文档教程](../../README_ch.md#文档教程),正式开启PaddleOCR的应用之旅。