Tingquan commited on
Commit
80b3ae9
·
verified ·
1 Parent(s): 58ccebe

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +176 -0
README.md ADDED
@@ -0,0 +1,176 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ ---
4
+
5
+ # en_PP-OCRv3_mobile_rec
6
+
7
+ ## Introduction
8
+
9
+ en_PP-OCRv3_mobile_rec is a text line recognition model within the PP-OCRv3_rec series, developed by the PaddleOCR team. The en_PP-OCRv3_mobile_rec model is an English-specific model trained based on PP-OCRv3_mobile_rec, and it supports English recognition. The key accuracy metrics are as follow:
10
+
11
+ <table>
12
+ <tr>
13
+ <th>Model</th>
14
+ <th>Recognition Avg Accuracy(%)</th>
15
+ <th>GPU Inference Time (ms)<br/>[Normal Mode / High-Performance Mode]</th>
16
+ <th>CPU Inference Time (ms)<br/>[Normal Mode / High-Performance Mode]</th>
17
+ <th>Model Storage Size (M)</th>
18
+ <th>Introduction</th>
19
+ </tr>
20
+ <tr>
21
+ <td>en_PP-OCRv3_mobile_rec</td>
22
+ <td>70.69</td>
23
+ <td>5.44 / 0.75</td>
24
+ <td>8.65 / 5.57</td>
25
+ <td>7.8 M </td>
26
+ <td>An ultra-lightweight English recognition model trained based on the PP-OCRv3 recognition model, supporting English and numeric character recognition.</td>
27
+ </tr>
28
+
29
+ </table>
30
+
31
+
32
+
33
+ **Note**: If any character (including punctuation) in a line is incorrect, the entire line is marked as wrong. This ensures higher accuracy in practical applications.
34
+
35
+ ## Quick Start
36
+
37
+ ### Installation
38
+
39
+ 1. PaddlePaddle
40
+
41
+ Please refer to the following commands to install PaddlePaddle using pip:
42
+
43
+ ```bash
44
+ # for CUDA11.8
45
+ python -m pip install paddlepaddle-gpu==3.0.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/
46
+
47
+ # for CUDA12.6
48
+ python -m pip install paddlepaddle-gpu==3.0.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/
49
+
50
+ # for CPU
51
+ python -m pip install paddlepaddle==3.0.0 -i https://www.paddlepaddle.org.cn/packages/stable/cpu/
52
+ ```
53
+
54
+ For details about PaddlePaddle installation, please refer to the [PaddlePaddle official website](https://www.paddlepaddle.org.cn/en/install/quick).
55
+
56
+ 2. PaddleOCR
57
+
58
+ Install the latest version of the PaddleOCR inference package from PyPI:
59
+
60
+ ```bash
61
+ python -m pip install paddleocr
62
+ ```
63
+
64
+ ### Model Usage
65
+
66
+ You can quickly experience the functionality with a single command:
67
+
68
+ ```bash
69
+ paddleocr text_recognition \
70
+ --model_name en_PP-OCRv3_mobile_rec \
71
+ -i https://cdn-uploads.huggingface.co/production/uploads/681c1ecd9539bdde5ae1733c/QmaPtftqwOgCtx0AIvU2z.png
72
+ ```
73
+
74
+ You can also integrate the model inference of the text recognition module into your project. Before running the following code, please download the sample image to your local machine.
75
+
76
+ ```python
77
+ from paddleocr import TextRecognition
78
+ model = TextRecognition(model_name="en_PP-OCRv3_mobile_rec")
79
+ output = model.predict(input="QmaPtftqwOgCtx0AIvU2z.png", batch_size=1)
80
+ for res in output:
81
+ res.print()
82
+ res.save_to_img(save_path="./output/")
83
+ res.save_to_json(save_path="./output/res.json")
84
+ ```
85
+
86
+ After running, the obtained result is as follows:
87
+
88
+ ```json
89
+ {'res': {'input_path': '/root/.paddlex/predict_input/QmaPtftqwOgCtx0AIvU2z.png', 'page_index': None, 'rec_text': 'the number of model parameters and FLOPs get larger, it', 'rec_score': 0.990352988243103}}
90
+ ```
91
+
92
+ The visualized image is as follows:
93
+
94
+ ![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/681c1ecd9539bdde5ae1733c/aT_U3ssYetaz4MiYfIwfZ.png)
95
+
96
+ For details about usage command and descriptions of parameters, please refer to the [Document](https://paddlepaddle.github.io/PaddleOCR/latest/en/version3.x/module_usage/text_recognition.html#iii-quick-start).
97
+
98
+ ### Pipeline Usage
99
+
100
+ The ability of a single model is limited. But the pipeline consists of several models can provide more capacity to resolve difficult problems in real-world scenarios.
101
+
102
+ #### PP-OCRv3
103
+
104
+ The general OCR pipeline is used to solve text recognition tasks by extracting text information from images and outputting it in string format. And there are 5 modules in the pipeline:
105
+ * Document Image Orientation Classification Module (Optional)
106
+ * Text Image Unwarping Module (Optional)
107
+ * Text Line Orientation Classification Module (Optional)
108
+ * Text Detection Module
109
+ * Text Recognition Module
110
+
111
+ Run a single command to quickly experience the OCR pipeline:
112
+
113
+ ```bash
114
+ paddleocr ocr -i https://cdn-uploads.huggingface.co/production/uploads/681c1ecd9539bdde5ae1733c/c3hSldnYVQXp48T5V0Ze4.png \
115
+ --text_recognition_model_name en_PP-OCRv3_mobile_rec \
116
+ --use_doc_orientation_classify False \
117
+ --use_doc_unwarping False \
118
+ --use_textline_orientation True \
119
+ --save_path ./output \
120
+ --device gpu:0
121
+ ```
122
+
123
+ Results are printed to the terminal:
124
+
125
+ ```json
126
+ {'res': {'input_path': '/root/.paddlex/predict_input/c3hSldnYVQXp48T5V0Ze4.png', 'page_index': None, 'model_settings': {'use_doc_preprocessor': True, 'use_textline_orientation': True}, 'doc_preprocessor_res': {'input_path': None, 'page_index': None, 'model_settings': {'use_doc_orientation_classify': False, 'use_doc_unwarping': False}, 'angle': -1}, 'dt_polys': array([[[252, 172],
127
+ ...,
128
+ [254, 241]],
129
+
130
+ ...,
131
+
132
+ [[665, 566],
133
+ ...,
134
+ [663, 601]]], dtype=int16), 'text_det_params': {'limit_side_len': 64, 'limit_type': 'min', 'thresh': 0.3, 'max_side_limit': 4000, 'box_thresh': 0.6, 'unclip_ratio': 1.5}, 'text_type': 'general', 'textline_orientation_angles': array([0, ..., 0]), 'text_rec_score_thresh': 0.0, 'rec_texts': ['The moon tells the sky', 'The sky tells the sea', 'The sea tells the tide', 'And the tide tells me', 'Lemn Sissay'], 'rec_scores': array([0.99890286, ..., 0.99840254]), 'rec_polys': array([[[252, 172],
135
+ ...,
136
+ [254, 241]],
137
+
138
+ ...,
139
+
140
+ [[665, 566],
141
+ ...,
142
+ [663, 601]]], dtype=int16), 'rec_boxes': array([[252, ..., 241],
143
+ ...,
144
+ [663, ..., 612]], dtype=int16)}}
145
+ ```
146
+
147
+ If save_path is specified, the visualization results will be saved under `save_path`. The visualization output is shown below:
148
+
149
+ ![image/jpeg](https://cdn-uploads.huggingface.co/production/uploads/681c1ecd9539bdde5ae1733c/PPUZ6f72Op5zCyq7T3ZKV.png)
150
+
151
+ The command-line method is for quick experience. For project integration, also only a few codes are needed as well:
152
+
153
+ ```python
154
+ from paddleocr import PaddleOCR
155
+
156
+ ocr = PaddleOCR(
157
+ text_recognition_model_name="en_PP-OCRv3_mobile_rec",
158
+ use_doc_orientation_classify=False, # Use use_doc_orientation_classify to enable/disable document orientation classification model
159
+ use_doc_unwarping=False, # Use use_doc_unwarping to enable/disable document unwarping module
160
+ use_textline_orientation=True, # Use use_textline_orientation to enable/disable textline orientation classification model
161
+ device="gpu:0", # Use device to specify GPU for model inference
162
+ )
163
+ result = ocr.predict("https://cdn-uploads.huggingface.co/production/uploads/681c1ecd9539bdde5ae1733c/c3hSldnYVQXp48T5V0Ze4.png")
164
+ for res in result:
165
+ res.print()
166
+ res.save_to_img("output")
167
+ res.save_to_json("output")
168
+ ```
169
+
170
+ The default model used in pipeline is `PP-OCRv5_server_rec`, so it is needed that specifing to `en_PP-OCRv3_mobile_rec` by argument `text_recognition_model_name`. And you can also use the local model file by argument `text_recognition_model_dir`. For details about usage command and descriptions of parameters, please refer to the [Document](https://paddlepaddle.github.io/PaddleOCR/latest/en/version3.x/pipeline_usage/OCR.html#2-quick-start).
171
+
172
+ ## Links
173
+
174
+ [PaddleOCR Repo](https://github.com/paddlepaddle/paddleocr)
175
+
176
+ [PaddleOCR Documentation](https://paddlepaddle.github.io/PaddleOCR/latest/en/index.html)