The Document Image Orientation Classification Module is primarily designed to distinguish the orientation of document images and correct them through post-processing. During processes such as document scanning or ID photo capturing, the device might be rotated to achieve clearer images, resulting in images with various orientations. Standard OCR pipelines may not handle these images effectively. By leveraging image classification techniques, the orientation of documents or IDs containing text regions can be pre-determined and adjusted, thereby improving the accuracy of OCR processing. The key accuracy metrics are as follow:
Model
Recognition Avg Accuracy(%)
Model Storage Size (M)
Introduction
PP-LCNet_x1_0_doc_ori
99.06
7
A document image classification model based on PP-LCNet_x1_0, with four categories: 0°, 90°, 180°, and 270°.
Quick Start
Installation
PaddlePaddle
Please refer to the following commands to install PaddlePaddle using pip:
bash
1# for CUDA11.82python -m pip install paddlepaddle-gpu==3.0.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu118/
34# for CUDA12.65python -m pip install paddlepaddle-gpu==3.0.0 -i https://www.paddlepaddle.org.cn/packages/stable/cu126/
67# for CPU8python -m pip installpaddlepaddle==3.0.0 -i https://www.paddlepaddle.org.cn/packages/stable/cpu/
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.
python
1from paddleocr import DocImgOrientationClassification
2model = DocImgOrientationClassification(model_name="PP-LCNet_x1_0_doc_ori")3output = model.predict(input="4ifXaBJmFByG_mAnF86Vv.png", batch_size=1)4for res in output:5 res.print()6 res.save_to_img(save_path="./output/")7 res.save_to_json(save_path="./output/res.json")
For details about usage command and descriptions of parameters, please refer to the Document.
Pipeline Usage
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.
doc_preprocessor
The Document Image Preprocessing Pipeline integrates two key functions: document orientation classification and geometric distortion correction. The document orientation classification module automatically identifies the four possible orientations of a document (0°, 90°, 180°, 270°), ensuring that the document is processed in the correct direction. The text image unwarping model is designed to correct geometric distortions that occur during document photography or scanning, restoring the document's original shape and proportions. This pipeline is suitable for digital document management, preprocessing tasks for OCR, and any scenario requiring improved document image quality. By automating orientation correction and geometric distortion correction, this module significantly enhances the accuracy and efficiency of document processing, providing a more reliable foundation for image analysis. The pipeline also offers flexible service-oriented deployment options, supporting calls from various programming languages on multiple hardware platforms. Additionally, the pipeline supports secondary development, allowing you to fine-tune the models on your own datasets and seamlessly integrate the trained models. And there are 2 modules in the pipeline:
If save_path is specified, the visualization results will be saved under save_path. The visualization output is shown below:
image/jpeg
The command-line method is for quick experience. For project integration, also only a few codes are needed as well:
python
1from paddleocr import DocPreprocessor
23ocr = DocPreprocessor(4 doc_orientation_classify_model_name="PP-LCNet_x1_0_doc_ori",5 use_doc_orientation_classify=True,# Use use_doc_orientation_classify to enable/disable document orientation classification model6 use_doc_unwarping=True,# Use use_doc_unwarping to enable/disable document unwarping module7 device="gpu:0",# Use device to specify GPU for model inference8)9result = ocr.predict("https://cdn-uploads.huggingface.co/production/uploads/681c1ecd9539bdde5ae1733c/pY6sY6wLDuoHF1-cGUvDr.png")10for res in result:11 res.print()12 res.save_to_img("output")13 res.save_to_json("output")
