rapid-orientation 0.0.5

Tools for classifying the direction of images containing text based ONNXRuntime.

rapid-orientation

1. Install package by pypi.

$ pip install rapid-orientation

2. Run by script.

• RapidOrientation has the default model_path value, you can set the different value of model_path to use different models, e.g. orientation_engine = RapidOrientation(model_path='rapid_orientation.onnx')
• See details, for README_Layout .
• 📌 layout.png source: link

import cv2
from rapid_orientation import RapidOrientation

orientation_engine = RapidOrientation()

img = cv2.imread('test_images/layout.png')

orientation_res, elapse = orientation_engine(img)
print(orientation_res)

3. Run by command line.

• Usage:

  $ 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
  

• Example:

  $ rapid_orientation -img layout.png
  

4. Result.

# Return str, four types:：0 | 90 | 180 | 270