7.2 KiB
7.2 KiB
Quick start of Chinese OCR model
1. Prepare for the environment
Please refer to quick installation to configure the PaddleOCR operating environment.
- Note: Support the use of PaddleOCR through whl package installation,pelease refer PaddleOCR Package.
2.inference models
The detection and recognition models on the mobile and server sides are as follows. For more models (including multiple languages), please refer to PP-OCR v1.1 series model list
Model introduction | Model name | Recommended scene | Detection model | Direction Classifier | Recognition model |
---|---|---|---|---|---|
Ultra-lightweight Chinese OCR model(8.1M) | ch_ppocr_mobile_v1.1_xx | Mobile-side/Server-side | inference model / pretrained model | inference model / pretrained model | inference model / pretrained model |
Universal Chinese OCR model(155.1M) | ch_ppocr_server_v1.1_xx | Server-side | inference model / pretrained model | inference model / pretrained model | inference model / pretrained model |
- If
wget
is not installed in the windows environment, you can copy the link to the browser to download when downloading the model, then uncompress it and place it in the corresponding directory.
Copy the download address of the inference model
for detection and recognition in the table above, and uncompress them.
mkdir inference && cd inference
# Download the detection model and unzip
wget {url/of/detection/inference_model} && tar xf {name/of/detection/inference_model/package}
# Download the recognition model and unzip
wget {url/of/recognition/inference_model} && tar xf {name/of/recognition/inference_model/package}
# Download the direction classifier model and unzip
wget {url/of/classification/inference_model} && tar xf {name/of/classification/inference_model/package}
cd ..
Take the ultra-lightweight model as an example:
mkdir inference && cd inference
# Download the detection model of the ultra-lightweight Chinese OCR model and uncompress it
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
# Download the recognition model of the ultra-lightweight Chinese OCR model and uncompress it
wget https://paddleocr.bj.bcebos.com/20-09-22/mobile/rec/ch_ppocr_mobile_v1.1_rec_infer.tar && tar xf ch_ppocr_mobile_v1.1_rec_infer.tar
# Download the direction classifier model of the ultra-lightweight Chinese OCR model and uncompress it
wget https://paddleocr.bj.bcebos.com/20-09-22/cls/ch_ppocr_mobile_v1.1_cls_infer.tar && tar xf ch_ppocr_mobile_v1.1_cls_infer.tar
cd ..
After decompression, the file structure should be as follows:
|-inference
|-ch_ppocr_mobile_v1.1_det_infer
|- model
|- params
|-ch_ppocr_mobile_v1.1_rec_infer
|- model
|- params
|-ch_ppocr_mobile_v1.1_cls_infer
|- model
|- params
...
3. Single image or image set prediction
- The following code implements text detection and recognition process. When performing prediction, you need to specify the path of a single image or image set through the parameter
image_dir
, the parameterdet_model_dir
specifies the path to detect the inference model, the parameterrec_model_dir
specifies the path to identify the inference model, the parameteruse_angle_cls
specifies whether to use the direction classifier, the parametercls_model_dir
specifies the path to identify the direction classifier model, the parameteruse_space_char
specifies whether to predict the space char. The visual results are saved to the./inference_results
folder by default.
# Predict a single image specified by image_dir
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
# Predict imageset specified by image_dir
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
# If you want to use the CPU for prediction, you need to set the use_gpu parameter to False
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
- Universal Chinese OCR model
Please follow the above steps to download the corresponding models and update the relevant parameters, The example is as follows.
# Predict a single image specified by image_dir
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
- Note
- If you want to use the recognition model which does not support space char recognition, please update the source code to the latest version and add parameters
--use_space_char=False
. - If you do not want to use direction classifier, please update the source code to the latest version and add parameters
--use_angle_cls=False
.
- If you want to use the recognition model which does not support space char recognition, please update the source code to the latest version and add parameters
- For more text detection and recognition tandem reasoning, please refer to the document tutorial
- Inference with Python inference engine。
In addition, the tutorial also provides other deployment methods for the Chinese OCR model: