📣 原有RapidStructure仓库已经移到RapidDoc下了,RapidStructure也将以RapidDoc方式重生
该部分主要是做含文字图像方向分类模型。模型来源:PaddleClas 含文字图像方向分类模型
模型类型 | 模型名称 | 模型大小 | 支持种类 |
---|---|---|---|
四方向分类 | rapid_orientation.onnx |
6.5M | 0 90 180 270 |
模型手动下载位置:link
由于模型较小,已经将分类模型(rapid_orientation.onnx
)打包进了whl包内:
pip install rapid-orientation
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