Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

RapidAI/RapidOrientation

Open more actions menu

Repository files navigation

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

Rapid Orientation

PyPI SemVer2.0

简介和说明

该部分主要是做含文字图像方向分类模型。模型来源: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

Releases

Sponsor this project

Used by

Contributors

Languages

Morty Proxy This is a proxified and sanitized view of the page, visit original site.