RapidAI/RapidOrientation

文档方向分类

📣 原有RapidStructure仓库已经移到RapidDoc下了，RapidStructure也将以RapidDoc方式重生

Rapid Orientation

简介和说明

该部分主要是做含文字图像方向分类模型。模型来源：PaddleClas 含文字图像方向分类模型

模型类型	模型名称	模型大小	支持种类
四方向分类	rapid_orientation.onnx	6.5M	0 90 180 270

安装

模型手动下载位置：link

由于模型较小，已经将分类模型(rapid_orientation.onnx)打包进了whl包内：

pip install rapid-orientation

Python脚本运行

import cv2

from rapid_orientation import RapidOrientation

orientation_engine = RapidOrientation()
img = cv2.imread("tests/test_files/img_rot180_demo.jpg")
cls_result, _ = orientation_engine(img)
print(cls_result)

终端运行

用法:

$ rapid_orientation -h
usage: rapid_orientation [-h] -img IMG_PATH [-m MODEL_PATH]

optional arguments:
-h, --help            show this help message and exit
-img IMG_PATH, --img_path IMG_PATH
                      Path to image for layout.
-m MODEL_PATH, --model_path MODEL_PATH
                      The model path used for inference.

示例:

rapid_orientation -img test_images/layout.png

结果

# 返回结果为str类型，有四类：0 | 90 | 180 | 270