onnxtr 0.6.0__py3-none-any.whl → 0.6.3__py3-none-any.whl

onnxtr/contrib/artefacts.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/contrib/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/file_utils.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/io/elements.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -331,7 +331,7 @@ class Page(Element):
         )
         # Create the body
         body = SubElement(page_hocr, "body")
-        SubElement(
+        page_div = SubElement(
             body,
             "div",
             attrib={
@@ -346,7 +346,7 @@ class Page(Element):
                 raise TypeError("XML export is only available for straight bounding boxes for now.")
             (xmin, ymin), (xmax, ymax) = block.geometry
             block_div = SubElement(
-                body,
+                page_div,
                 "div",
                 attrib={
                     "class": "ocr_carea",

onnxtr/io/html.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/io/image.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/io/pdf.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/io/reader.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/_utils.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/builder.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/classification/models/mobilenet.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -24,16 +24,16 @@ default_cfgs: dict[str, dict[str, Any]] = {
         "std": (0.299, 0.296, 0.301),
         "input_shape": (3, 256, 256),
         "classes": [0, -90, 180, 90],
-        "url": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.0.1/mobilenet_v3_small_crop_orientation-5620cf7e.onnx",
-        "url_8_bit": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.1.2/mobilenet_v3_small_crop_orientation_static_8_bit-4cfaa621.onnx",
+        "url": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.6.0/mobilenet_v3_small_crop_orientation-4fde60a1.onnx",
+        "url_8_bit": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.6.0/mobilenet_v3_small_crop_orientation_static_8_bit-c32c7721.onnx",
     },
     "mobilenet_v3_small_page_orientation": {
         "mean": (0.694, 0.695, 0.693),
         "std": (0.299, 0.296, 0.301),
         "input_shape": (3, 512, 512),
         "classes": [0, -90, 180, 90],
-        "url": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.0.1/mobilenet_v3_small_page_orientation-d3f76d79.onnx",
-        "url_8_bit": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.1.2/mobilenet_v3_small_page_orientation_static_8_bit-3e5ef3dc.onnx",
+        "url": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.6.0/mobilenet_v3_small_page_orientation-60606ce4.onnx",
+        "url_8_bit": "https://github.com/felixdittrich92/OnnxTR/releases/download/v0.6.0/mobilenet_v3_small_page_orientation_static_8_bit-13b5b014.onnx",
     },
 }
 

onnxtr/models/classification/predictor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/classification/zoo.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/detection/_utils/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/detection/core.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/detection/models/differentiable_binarization.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/detection/models/fast.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/detection/models/linknet.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/detection/postprocessor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -9,7 +9,6 @@
 import cv2
 import numpy as np
 import pyclipper
-from shapely.geometry import Polygon
 
 from ..core import DetectionPostProcessor
 
@@ -54,9 +53,8 @@ class GeneralDetectionPostProcessor(DetectionPostProcessor):
             area = (rect[1][0] + 1) * (1 + rect[1][1])
             length = 2 * (rect[1][0] + rect[1][1]) + 2
         else:
-            poly = Polygon(points)
-            area = poly.area
-            length = poly.length
+            area = cv2.contourArea(points)
+            length = cv2.arcLength(points, closed=True)
         distance = area * self.unclip_ratio / length  # compute distance to expand polygon
         offset = pyclipper.PyclipperOffset()
         offset.AddPath(points, pyclipper.JT_ROUND, pyclipper.ET_CLOSEDPOLYGON)

onnxtr/models/detection/predictor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/detection/zoo.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -44,6 +44,7 @@ def _predictor(
             raise ValueError(f"unknown architecture: {type(arch)}")
 
         _model = arch
+        _model.assume_straight_pages = assume_straight_pages
         _model.postprocessor.assume_straight_pages = assume_straight_pages
 
     kwargs["mean"] = kwargs.get("mean", _model.cfg["mean"])

onnxtr/models/engine.py

@@ -1,9 +1,10 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
 
 import logging
+import os
 from typing import Any
 
 import numpy as np
@@ -15,6 +16,9 @@ from onnxruntime import (
     get_available_providers,
     get_device,
 )
+from onnxruntime.capi._pybind_state import set_default_logger_severity
+
+set_default_logger_severity(int(os.getenv("ORT_LOG_SEVERITY_LEVEL", 4)))
 
 from onnxtr.utils.data import download_from_url
 from onnxtr.utils.geometry import shape_translate
@@ -90,6 +94,8 @@ class Engine:
     def __init__(self, url: str, engine_cfg: EngineConfig | None = None, **kwargs: Any) -> None:
         engine_cfg = engine_cfg if isinstance(engine_cfg, EngineConfig) else EngineConfig()
         archive_path = download_from_url(url, cache_subdir="models", **kwargs) if "http" in url else url
+        # NOTE: older onnxruntime versions require a string path for windows
+        archive_path = rf"{archive_path}"
         # Store model path for each model
         self.model_path = archive_path
         self.session_options = engine_cfg.session_options

onnxtr/models/factory/hub.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/predictor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/predictor/predictor.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/preprocessor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/recognition/core.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/recognition/models/crnn.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/recognition/models/master.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/recognition/models/parseq.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/recognition/models/sar.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/recognition/models/vitstr.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/recognition/predictor/_utils.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/recognition/predictor/base.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/recognition/utils.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/recognition/zoo.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/models/zoo.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/transforms/base.py

@@ -1,9 +1,11 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
 
 
+import math
+
 import numpy as np
 from PIL import Image, ImageOps
 
@@ -37,37 +39,51 @@ class Resize:
             raise AssertionError("size should be either a tuple or an int")
 
     def __call__(self, img: np.ndarray) -> np.ndarray:
-        img = (img * 255).astype(np.uint8) if img.dtype != np.uint8 else img
-        h, w = img.shape[:2] if img.ndim == 3 else img.shape[1:3]
+        if img.dtype != np.uint8:
+            img_pil = Image.fromarray((img * 255).clip(0, 255).astype(np.uint8))
+        else:
+            img_pil = Image.fromarray(img)
+
         sh, sw = self.size
+        w, h = img_pil.size
 
         if not self.preserve_aspect_ratio:
-            return np.array(Image.fromarray(img).resize((sw, sh), resample=self.interpolation))
+            img_resized_pil = img_pil.resize((sw, sh), resample=self.interpolation)
+            return np.array(img_resized_pil)
 
         actual_ratio = h / w
         target_ratio = sh / sw
 
-        if target_ratio == actual_ratio:
-            return np.array(Image.fromarray(img).resize((sw, sh), resample=self.interpolation))
-
         if actual_ratio > target_ratio:
-            tmp_size = (int(sh / actual_ratio), sh)
+            new_h = sh
+            new_w = max(int(sh / actual_ratio), 1)
         else:
-            tmp_size = (sw, int(sw * actual_ratio))
+            new_w = sw
+            new_h = max(int(sw * actual_ratio), 1)
 
-        img_resized = Image.fromarray(img).resize(tmp_size, resample=self.interpolation)
-        pad_left = pad_top = 0
-        pad_right = sw - img_resized.width
-        pad_bottom = sh - img_resized.height
+        img_resized_pil = img_pil.resize((new_w, new_h), resample=self.interpolation)
+
+        delta_w = sw - new_w
+        delta_h = sh - new_h
 
         if self.symmetric_pad:
-            pad_left = pad_right // 2
-            pad_right -= pad_left
-            pad_top = pad_bottom // 2
-            pad_bottom -= pad_top
+            # Symmetric padding
+            pad_left = math.ceil(delta_w / 2)
+            pad_right = math.floor(delta_w / 2)
+            pad_top = math.ceil(delta_h / 2)
+            pad_bottom = math.floor(delta_h / 2)
+        else:
+            # Asymmetric padding
+            pad_left, pad_top = 0, 0
+            pad_right, pad_bottom = delta_w, delta_h
+
+        img_padded_pil = ImageOps.expand(
+            img_resized_pil,
+            border=(pad_left, pad_top, pad_right, pad_bottom),
+            fill=0,
+        )
 
-        img_resized = ImageOps.expand(img_resized, (pad_left, pad_top, pad_right, pad_bottom))
-        return np.array(img_resized)
+        return np.array(img_padded_pil)
 
     def __repr__(self) -> str:
         interpolate_str = self.interpolation

onnxtr/utils/common_types.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/utils/data.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/utils/fonts.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/utils/geometry.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/utils/multithreading.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/utils/reconstitution.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/utils/repr.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.

onnxtr/utils/visualization.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -276,7 +276,11 @@ def draw_boxes(boxes: np.ndarray, image: np.ndarray, color: tuple[int, int, int]
     for box in _boxes.tolist():
         xmin, ymin, xmax, ymax = box
         image = cv2.rectangle(
-            image, (xmin, ymin), (xmax, ymax), color=color if isinstance(color, tuple) else (0, 0, 255), thickness=2
+            image,
+            (xmin, ymin),
+            (xmax, ymax),
+            color=color if isinstance(color, tuple) else (0, 0, 255),
+            thickness=2,
         )
     plt.imshow(image)
     plt.plot(**kwargs)

onnxtr/utils/vocabs.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2024, Mindee | Felix Dittrich.
+# Copyright (C) 2021-2025, Mindee | Felix Dittrich.
 
 # This program is licensed under the Apache License 2.0.
 # See LICENSE or go to <https://opensource.org/licenses/Apache-2.0> for full license details.
@@ -9,32 +9,92 @@ __all__ = ["VOCABS"]
 
 
 VOCABS: dict[str, str] = {
-    "digits": string.digits,
-    "ascii_letters": string.ascii_letters,
-    "punctuation": string.punctuation,
-    "currency": "£€¥¢฿",
-    "ancient_greek": "αβγδεζηθικλμνξοπρστυφχψωΑΒΓΔΕΖΗΘΙΚΛΜΝΞΟΠΡΣΤΥΦΧΨΩ",
-    "arabic_letters": "ءآأؤإئابةتثجحخدذرزسشصضطظعغـفقكلمنهوىي",
-    "persian_letters": "پچڢڤگ",
-    "arabic_digits": "٠١٢٣٤٥٦٧٨٩",
+    # Arabic & Persian
     "arabic_diacritics": "ًٌٍَُِّْ",
+    "arabic_digits": "٠١٢٣٤٥٦٧٨٩",
+    "arabic_letters": "ءآأؤإئابةتثجحخدذرزسشصضطظعغـفقكلمنهوىي",
     "arabic_punctuation": "؟؛«»—",
-    "hindi_letters": "अआइईउऊऋॠऌॡएऐओऔअंअःकखगघङचछजझञटठडढणतथदधनपफबभमयरलवशषसह",
-    "hindi_digits": "०१२३४५६७८९",
-    "hindi_punctuation": "।,?!:्ॐ॰॥॰",
-    "bangla_letters": "অআইঈউঊঋএঐওঔকখগঘঙচছজঝঞটঠডঢণতথদধনপফবভমযরলশষসহ়ঽািীুূৃেৈোৌ্ৎংঃঁ",
+    "persian_letters": "پچڢڤگ",
+    # Bangla
     "bangla_digits": "০১২৩৪৫৬৭৮৯",
+    "bangla_letters": "অআইঈউঊঋএঐওঔকখগঘঙচছজঝঞটঠডঢণতথদধনপফবভমযরলশষসহ়ঽািীুূৃেৈোৌ্ৎংঃঁ",
+    # Cyrillic
     "generic_cyrillic_letters": "абвгдежзийклмнопрстуфхцчшщьюяАБВГДЕЖЗИЙКЛМНОПРСТУФХЦЧШЩЬЮЯ",
+    "russian_cyrillic_letters": "ёыэЁЫЭ",
+    "russian_signs": "ъЪ",
+    # Greek
+    "ancient_greek": "αβγδεζηθικλμνξοπρστυφχψωΑΒΓΔΕΖΗΘΙΚΛΜΝΞΟΠΡΣΤΥΦΧΨΩ",
+    # Gujarati
+    "gujarati_consonants": "ખગઘચછજઝઞટઠડઢણતથદધનપફબભમયરલવશસહળક્ષ",
+    "gujarati_digits": "૦૧૨૩૪૫૬૭૮૯",
+    "gujarati_punctuation": "૰ઽ◌ંઃ॥ૐ઼ઁ" + "૱",
+    "gujarati_vowels": "અઆઇઈઉઊઋએઐઓ",
+    # Hindi
+    "hindi_digits": "०१२३४५६७८९",
+    "hindi_letters": "अआइईउऊऋॠऌॡएऐओऔंःकखगघङचछजझञटठडढणतथदधनपफबभमयरलवशषसह",
+    "hindi_punctuation": "।,?!:्ॐ॰॥",
+    # Hebrew
+    "hebrew_cantillations": "֑֖֛֢֣֤֥֦֧֪֚֭֮֒֓֔֕֗֘֙֜֝֞֟֠֡֨֩֫֬֯",
+    "hebrew_letters": "אבגדהוזחטיךכלםמןנסעףפץצקרשת",
+    "hebrew_specials": "ׯװױײיִﬞײַﬠﬡﬢﬣﬤﬥﬦﬧﬨ﬩שׁשׂשּׁשּׂאַאָאּבּגּדּהּוּזּטּיּךּכּלּמּנּסּףּפּצּקּרּשּתּוֹבֿכֿפֿﭏ",
+    "hebrew_punctuation": "ֽ־ֿ׀ׁׂ׃ׅׄ׆׳״",
+    "hebrew_vowels": "ְֱֲֳִֵֶַָׇֹֺֻ",
+    # Latin
+    "digits": string.digits,
+    "ascii_letters": string.ascii_letters,
+    "punctuation": string.punctuation,
+    "currency": "£€¥¢฿",
 }
 
+# Latin & latin-dependent alphabets
 VOCABS["latin"] = VOCABS["digits"] + VOCABS["ascii_letters"] + VOCABS["punctuation"]
 VOCABS["english"] = VOCABS["latin"] + "°" + VOCABS["currency"]
-VOCABS["legacy_french"] = VOCABS["latin"] + "°" + "àâéèêëîïôùûçÀÂÉÈËÎÏÔÙÛÇ" + VOCABS["currency"]
+
+VOCABS["czech"] = VOCABS["english"] + "áčďéěíňóřšťúůýžÁČĎÉĚÍŇÓŘŠŤÚŮÝŽ"
+
+VOCABS["danish"] = VOCABS["english"] + "æøåÆØÅ"
+
+VOCABS["dutch"] = VOCABS["english"] + "áéíóúüñÁÉÍÓÚÜÑ"
+
 VOCABS["french"] = VOCABS["english"] + "àâéèêëîïôùûüçÀÂÉÈÊËÎÏÔÙÛÜÇ"
+VOCABS["legacy_french"] = VOCABS["latin"] + "°" + "àâéèêëîïôùûçÀÂÉÈËÎÏÔÙÛÇ" + VOCABS["currency"]
+
+VOCABS["finnish"] = VOCABS["english"] + "äöÄÖ"
+
+VOCABS["german"] = VOCABS["english"] + "äöüßÄÖÜẞ"
+
+VOCABS["croatian"] = VOCABS["english"] + "ČčĆćĐ𩹮ž"
+
+VOCABS["hebrew"] = (
+    VOCABS["english"]
+    + VOCABS["hebrew_letters"]
+    + VOCABS["hebrew_vowels"]
+    + VOCABS["hebrew_punctuation"]
+    + VOCABS["hebrew_cantillations"]
+    + VOCABS["hebrew_specials"]
+    + "₪"
+)
+
+VOCABS["italian"] = VOCABS["english"] + "àèéìíîòóùúÀÈÉÌÍÎÒÓÙÚ"
+
+VOCABS["norwegian"] = VOCABS["english"] + "æøåÆØÅ"
+
+VOCABS["polish"] = VOCABS["english"] + "ąćęłńóśźżĄĆĘŁŃÓŚŹŻ"
+
 VOCABS["portuguese"] = VOCABS["english"] + "áàâãéêíïóôõúüçÁÀÂÃÉÊÍÏÓÔÕÚÜÇ"
+
 VOCABS["spanish"] = VOCABS["english"] + "áéíóúüñÁÉÍÓÚÜÑ" + "¡¿"
-VOCABS["italian"] = VOCABS["english"] + "àèéìíîòóùúÀÈÉÌÍÎÒÓÙÚ"
-VOCABS["german"] = VOCABS["english"] + "äöüßÄÖÜẞ"
+
+VOCABS["swedish"] = VOCABS["english"] + "åäöÅÄÖ"
+
+VOCABS["vietnamese"] = (
+    VOCABS["english"]
+    + "áàảạãăắằẳẵặâấầẩẫậđéèẻẽẹêếềểễệóòỏõọôốồổộỗơớờởợỡúùủũụưứừửữựíìỉĩịýỳỷỹỵ"
+    + "ÁÀẢẠÃĂẮẰẲẴẶÂẤẦẨẪẬĐÉÈẺẼẸÊẾỀỂỄỆÓÒỎÕỌÔỐỒỔỘỖƠỚỜỞỢỠÚÙỦŨỤƯỨỪỬỮỰÍÌỈĨỊÝỲỶỸỴ"
+)
+
+# Non-latin alphabets.
+# Arabic
 VOCABS["arabic"] = (
     VOCABS["digits"]
     + VOCABS["arabic_digits"]
@@ -44,24 +104,37 @@ VOCABS["arabic"] = (
     + VOCABS["arabic_punctuation"]
     + VOCABS["punctuation"]
 )
-VOCABS["czech"] = VOCABS["english"] + "áčďéěíňóřšťúůýžÁČĎÉĚÍŇÓŘŠŤÚŮÝŽ"
-VOCABS["polish"] = VOCABS["english"] + "ąćęłńóśźżĄĆĘŁŃÓŚŹŻ"
-VOCABS["dutch"] = VOCABS["english"] + "áéíóúüñÁÉÍÓÚÜÑ"
-VOCABS["norwegian"] = VOCABS["english"] + "æøåÆØÅ"
-VOCABS["danish"] = VOCABS["english"] + "æøåÆØÅ"
-VOCABS["finnish"] = VOCABS["english"] + "äöÄÖ"
-VOCABS["swedish"] = VOCABS["english"] + "åäöÅÄÖ"
-VOCABS["vietnamese"] = (
-    VOCABS["english"]
-    + "áàảạãăắằẳẵặâấầẩẫậđéèẻẽẹêếềểễệóòỏõọôốồổộỗơớờởợỡúùủũụưứừửữựiíìỉĩịýỳỷỹỵ"
-    + "ÁÀẢẠÃĂẮẰẲẴẶÂẤẦẨẪẬĐÉÈẺẼẸÊẾỀỂỄỆÓÒỎÕỌÔỐỒỔỘỖƠỚỜỞỢỠÚÙỦŨỤƯỨỪỬỮỰIÍÌỈĨỊÝỲỶỸỴ"
+
+# Bangla
+VOCABS["bangla"] = VOCABS["bangla_letters"] + VOCABS["bangla_digits"]
+
+# Gujarati
+VOCABS["gujarati"] = (
+    VOCABS["gujarati_vowels"]
+    + VOCABS["gujarati_consonants"]
+    + VOCABS["gujarati_digits"]
+    + VOCABS["gujarati_punctuation"]
+    + VOCABS["punctuation"]
 )
-VOCABS["hebrew"] = VOCABS["english"] + "אבגדהוזחטיכלמנסעפצקרשת" + "₪"
+
+# Hindi
 VOCABS["hindi"] = VOCABS["hindi_letters"] + VOCABS["hindi_digits"] + VOCABS["hindi_punctuation"]
-VOCABS["bangla"] = VOCABS["bangla_letters"] + VOCABS["bangla_digits"]
+
+# Cyrillic
+VOCABS["russian"] = (
+    VOCABS["generic_cyrillic_letters"]
+    + VOCABS["russian_cyrillic_letters"]
+    + VOCABS["russian_signs"]
+    + VOCABS["digits"]
+    + VOCABS["punctuation"]
+    + "₽"
+)
+
 VOCABS["ukrainian"] = (
     VOCABS["generic_cyrillic_letters"] + VOCABS["digits"] + VOCABS["punctuation"] + VOCABS["currency"] + "ґіїєҐІЇЄ₴"
 )
+
+# Multi-lingual
 VOCABS["multilingual"] = "".join(
     dict.fromkeys(
         VOCABS["french"]
@@ -69,6 +142,7 @@ VOCABS["multilingual"] = "".join(
         + VOCABS["spanish"]
         + VOCABS["german"]
         + VOCABS["czech"]
+        + VOCABS["croatian"]
         + VOCABS["polish"]
         + VOCABS["dutch"]
         + VOCABS["italian"]

onnxtr/version.py

@@ -1 +1 @@
-__version__ = 'v0.6.0'
+__version__ = 'v0.6.3'

{onnxtr-0.6.0.dist-info → onnxtr-0.6.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: onnxtr
-Version: 0.6.0
+Version: 0.6.3
 Summary: Onnx Text Recognition (OnnxTR): docTR Onnx-Wrapper for high-performance OCR on documents.
 Author-email: Felix Dittrich <felixdittrich92@gmail.com>
 Maintainer: Felix Dittrich
@@ -229,7 +229,6 @@ Requires-Dist: numpy<3.0.0,>=1.16.0
 Requires-Dist: scipy<2.0.0,>=1.4.0
 Requires-Dist: pypdfium2<5.0.0,>=4.11.0
 Requires-Dist: pyclipper<2.0.0,>=1.2.0
-Requires-Dist: shapely<3.0.0,>=1.6.0
 Requires-Dist: rapidfuzz<4.0.0,>=3.0.0
 Requires-Dist: langdetect<2.0.0,>=1.0.9
 Requires-Dist: huggingface-hub<1.0.0,>=0.23.0
@@ -264,6 +263,7 @@ Provides-Extra: testing
 Requires-Dist: pytest>=5.3.2; extra == "testing"
 Requires-Dist: coverage[toml]>=4.5.4; extra == "testing"
 Requires-Dist: requests>=2.20.0; extra == "testing"
+Requires-Dist: pytest-memray>=1.7.0; extra == "testing"
 Provides-Extra: quality
 Requires-Dist: ruff>=0.1.5; extra == "quality"
 Requires-Dist: mypy>=0.812; extra == "quality"
@@ -277,9 +277,11 @@ Requires-Dist: mplcursors>=0.3; extra == "dev"
 Requires-Dist: pytest>=5.3.2; extra == "dev"
 Requires-Dist: coverage[toml]>=4.5.4; extra == "dev"
 Requires-Dist: requests>=2.20.0; extra == "dev"
+Requires-Dist: pytest-memray>=1.7.0; extra == "dev"
 Requires-Dist: ruff>=0.1.5; extra == "dev"
 Requires-Dist: mypy>=0.812; extra == "dev"
 Requires-Dist: pre-commit>=2.17.0; extra == "dev"
+Dynamic: license-file
 
 <p align="center">
   <img src="https://github.com/felixdittrich92/OnnxTR/raw/main/docs/images/logo.jpg" width="40%">
@@ -290,9 +292,11 @@ Requires-Dist: pre-commit>=2.17.0; extra == "dev"
 [![codecov](https://codecov.io/gh/felixdittrich92/OnnxTR/graph/badge.svg?token=WVFRCQBOLI)](https://codecov.io/gh/felixdittrich92/OnnxTR)
 [![Codacy Badge](https://app.codacy.com/project/badge/Grade/4fff4d764bb14fb8b4f4afeb9587231b)](https://app.codacy.com/gh/felixdittrich92/OnnxTR/dashboard?utm_source=gh&utm_medium=referral&utm_content=&utm_campaign=Badge_grade)
 [![CodeFactor](https://www.codefactor.io/repository/github/felixdittrich92/onnxtr/badge)](https://www.codefactor.io/repository/github/felixdittrich92/onnxtr)
-[![Pypi](https://img.shields.io/badge/pypi-v0.6.0-blue.svg)](https://pypi.org/project/OnnxTR/)
+[![Socket Badge](https://socket.dev/api/badge/pypi/package/onnxtr/0.6.2?artifact_id=tar-gz)](https://socket.dev/pypi/package/onnxtr/overview/0.6.2/tar-gz)
+[![Pypi](https://img.shields.io/badge/pypi-v0.6.3-blue.svg)](https://pypi.org/project/OnnxTR/)
 [![Docker Images](https://img.shields.io/badge/Docker-4287f5?style=flat&logo=docker&logoColor=white)](https://github.com/felixdittrich92/OnnxTR/pkgs/container/onnxtr)
 [![Hugging Face Spaces](https://img.shields.io/badge/%F0%9F%A4%97%20Hugging%20Face-Spaces-blue)](https://huggingface.co/spaces/Felix92/OnnxTR-OCR)
+![PyPI - Downloads](https://img.shields.io/pypi/dm/onnxtr)
 
 > :warning: Please note that this is a wrapper around the [doctr](https://github.com/mindee/doctr) library to provide a Onnx pipeline for docTR. For feature requests, which are not directly related to the Onnx pipeline, please refer to the base project.
 

onnxtr-0.6.3.dist-info/RECORD

@@ -0,0 +1,75 @@
+onnxtr/__init__.py,sha256=h7Wc2tuHLsaoCk5xNpEFEK-g11A6SJA7nAasA76TQ_Y,100
+onnxtr/file_utils.py,sha256=rlflwU-KTHQ6wD1GU2EqPJs4y7gTyZSUnhtx9w2lIOk,1067
+onnxtr/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+onnxtr/version.py,sha256=59OmW64-lZAdyvZGKkVzWr6b7GEnjM7Dfx9gTahsuvc,23
+onnxtr/contrib/__init__.py,sha256=0Jo052I1DIdbNVYQBg213UJBNH3UKZIOcn2HVusQlj4,39
+onnxtr/contrib/artefacts.py,sha256=4uJyXrl2YEjeWaGC8pAVC5227eVzb34vTSbAe0fzGbw,5323
+onnxtr/contrib/base.py,sha256=SPPWd8SVwZr4kGPI-yM8UdQguEZ0awRBUZGjGKZEC-U,3135
+onnxtr/io/__init__.py,sha256=kS7tKGFvzxOCWBOun-Y8n9CsziwRKNynjwpZEUUI03M,106
+onnxtr/io/elements.py,sha256=F-AcKzVqDKVE5zfHEVGRhRGJWt8Ln54ndk3cVNQXmas,17751
+onnxtr/io/html.py,sha256=VleLE-94eJ2BermRZGDuKKLruE2p3bP1c3ZtaNkciF0,716
+onnxtr/io/image.py,sha256=ehOcXzH1eFjwLO3pwiV3bUwGcn4SdW-FO9DVbUjWw2M,1700
+onnxtr/io/pdf.py,sha256=cO_e9MWh2uCu20a9eLqp3y3GS9idpROkOuNkCU5yE8U,1327
+onnxtr/io/reader.py,sha256=JBDoGXaQsiV3eZxDu4Sd8aeULIEmLV5oNGqGXLAmtBk,2755
+onnxtr/models/__init__.py,sha256=QTfZlqUyv1d7NUCbGIUFM1DLOOXe-cqHZ7uaKkGdXvk,157
+onnxtr/models/_utils.py,sha256=m_6Qnx9QNIxmf_rrYmjd0htFmGa6ySZLoQ9kPjtMlpA,6424
+onnxtr/models/builder.py,sha256=aaMHLZl8fayeWsXHsvSy5Tq2yQWTO0gpj4Ob4kNZ0xk,14076
+onnxtr/models/engine.py,sha256=mtU4kKoWcgkcLELIJKTxYD2GMFAVZ1b_i7tSG2VeYQ8,5113
+onnxtr/models/zoo.py,sha256=44H7NfE4DWH7KgmF8vod9fMlPnxw41dwjpwjuIJkLHg,5303
+onnxtr/models/classification/__init__.py,sha256=h1bZs55iLJBMATtzS4ntTKwfD6OGXBiiqGv_hEnOFnE,41
+onnxtr/models/classification/zoo.py,sha256=4hD51Mdf5gkSoQ4g9zTBvurbBdrsaFFEl5WFpptBtNk,4271
+onnxtr/models/classification/models/__init__.py,sha256=rohbM6ZQslfYchi7feZwwh-sX3XXRUhgtEJQeurAytQ,24
+onnxtr/models/classification/models/mobilenet.py,sha256=pRI97DvBpBE-h2VymUG6sw_Jn-9-dUdvIMLavYF1syk,4818
+onnxtr/models/classification/predictor/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
+onnxtr/models/classification/predictor/base.py,sha256=hpLYmhn8PCbx46zYBM8rZiy2vabpN3YkviwmmbVaqRo,2313
+onnxtr/models/detection/__init__.py,sha256=h1bZs55iLJBMATtzS4ntTKwfD6OGXBiiqGv_hEnOFnE,41
+onnxtr/models/detection/core.py,sha256=8avDAHBh-S3jjQ-4WTcbXQvUgwrvJtFPti2ghUtE9uA,3497
+onnxtr/models/detection/zoo.py,sha256=RbzG9nR0Oa9bV0vwEBer2GDJMg1XvthpRLBxD2MsUvQ,3385
+onnxtr/models/detection/_utils/__init__.py,sha256=oPkIYbySSbLsOk02wVPNO9bUuywC47YjaenfyTwfOsw,20
+onnxtr/models/detection/_utils/base.py,sha256=MUn3cHUSq-dIKAA021ZUrTm0SLpAC9qsiw1BAWvcppc,2300
+onnxtr/models/detection/models/__init__.py,sha256=6Ea6knYrVCR2jAmPlsVWmCdHe-c6lSRETSAuZGfhx8I,85
+onnxtr/models/detection/models/differentiable_binarization.py,sha256=38LrxqtcVJfEdpeTu3aSeBOWmf1J_cLUDQyAfXVIlhc,6630
+onnxtr/models/detection/models/fast.py,sha256=8qsBwlV9J-eDqFDn0JMtvrvql3WmfWvKEy3iJ6_sSCY,6188
+onnxtr/models/detection/models/linknet.py,sha256=jgCOf9d7sQPNZ4qKM5l4O3PQFBnBOJJklNhcCXKDBrs,6666
+onnxtr/models/detection/postprocessor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+onnxtr/models/detection/postprocessor/base.py,sha256=sAjeYH3G_EMoN0yKOlk503ZGChVX85mNuhvBZj36PcI,5657
+onnxtr/models/detection/predictor/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
+onnxtr/models/detection/predictor/base.py,sha256=Yumq_zTSk5rnCbTJiVeOr1AQp6hv2yScI0MvXzl0n3w,2293
+onnxtr/models/factory/__init__.py,sha256=cKPoH2V2157lLMTR2zsljG3_IQHziodqR-XK_LG0D_I,19
+onnxtr/models/factory/hub.py,sha256=CzjFJ5ioPBG-QC8NYET51B-vzHn2t2EYj_tsoZXan44,7126
+onnxtr/models/predictor/__init__.py,sha256=XL25XkRkgyK7mldF-CWhg2MMakSdP5vLpDLwL59hphk,25
+onnxtr/models/predictor/base.py,sha256=1E70wZ6papp7OZkRCgY3Kf7xxMTpxIce_5ExFtT-GAg,9432
+onnxtr/models/predictor/predictor.py,sha256=QLldSoBBfX72hw5dricsmVMnb1QzwYnYdOAmkEEBFpc,6377
+onnxtr/models/preprocessor/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
+onnxtr/models/preprocessor/base.py,sha256=oGvc2yNZJjGqHE6HFJ-2k7Z5BJ6Dcw6IwS5iY12cy_E,3963
+onnxtr/models/recognition/__init__.py,sha256=h1bZs55iLJBMATtzS4ntTKwfD6OGXBiiqGv_hEnOFnE,41
+onnxtr/models/recognition/core.py,sha256=V6DMa3ABT0amr9vPSQSyo0dZqK-Y8qmA__XWhOxItSU,730
+onnxtr/models/recognition/utils.py,sha256=dtKsm5zycxy0t95AWKz9FnOQnQFiEXT6BZ94jXsecgA,3503
+onnxtr/models/recognition/zoo.py,sha256=HA9OhykB2539iIvInOPZez8nitj0fjF2eEgos-u8vjg,2857
+onnxtr/models/recognition/models/__init__.py,sha256=IXfiuzzkft8O1CpBZWYTpFw19y49mt5rJ_iGSdaWiU0,105
+onnxtr/models/recognition/models/crnn.py,sha256=9NwYTsrAalNHuNkxURe574v4tvKY-BsJOb25X4XDIKk,8786
+onnxtr/models/recognition/models/master.py,sha256=H_KcoJCK8nElyID5FKJdld5xapu5ZOWVcH5_H68_N1I,4669
+onnxtr/models/recognition/models/parseq.py,sha256=qe7lavwFT37SPLSMWoPaoCYo7yrTYGcCYqQraTwIuQY,4512
+onnxtr/models/recognition/models/sar.py,sha256=5v0VOPACvB_-I_kDh57pA00GmunIh6frD_4I6ojSjiQ,4523
+onnxtr/models/recognition/models/vitstr.py,sha256=U1nlfmoga4-8Lv9aHuDQiyidp5lNdFGAi5xui7N0cUs,5964
+onnxtr/models/recognition/predictor/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
+onnxtr/models/recognition/predictor/_utils.py,sha256=YW6g3zzmfX1fNTaQyDk483xj1JibtNSA6lr1cYdsqm0,3329
+onnxtr/models/recognition/predictor/base.py,sha256=FhpVKHnKDALssssEjYAy0IixNRoj1DQaJ1ncsF6_a1o,2488
+onnxtr/transforms/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
+onnxtr/transforms/base.py,sha256=jKYrJ99BQDrIgdPL-OSfDw8KoAQsW6Su2n0xLhxSsp8,4120
+onnxtr/utils/__init__.py,sha256=pESRJKtcQyjRxiMgZPhtPYeLbCj-YSGyMVRHTbcMONU,94
+onnxtr/utils/common_types.py,sha256=BiPSNYSU8p2R-LqBmyFw7TL-VN4RliKKlRwFOeYkpng,551
+onnxtr/utils/data.py,sha256=K7t5dvQoCWsxLUwOD3o70TxCFdwe7laCVGEmtF0XHMQ,4262
+onnxtr/utils/fonts.py,sha256=WAUitozOFoYZl6UI4uHEt3qkYvL_XIZ0NmEJvVvYP6U,1282
+onnxtr/utils/geometry.py,sha256=-5Rx6mJN3eRqyv2QqQPkwjKKQ-eA5Vu3wjLdgMLR2Qg,20418
+onnxtr/utils/multithreading.py,sha256=P8LXpaZSKRYsYZw6SRlWWXiQMDswk8zEgDSbCYVHzSA,1994
+onnxtr/utils/reconstitution.py,sha256=2SUkaReD5eJ5NFs7V7Z0m4N4lUr8QB7oH1mSHTC3vLA,6119
+onnxtr/utils/repr.py,sha256=lAcKRShXLpJ2DvsS2qf3vYq0lJrMJa30kLYXmfN3rV0,2105
+onnxtr/utils/visualization.py,sha256=MImlxVKOLXPriJYSrBfjZfJw0GEZ1gT8eni3KXZHrXk,9831
+onnxtr/utils/vocabs.py,sha256=Ad6EkGcb6-kiYsDHfTgrKoohKhosb_1jkb2mv7ifbiY,5834
+onnxtr-0.6.3.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+onnxtr-0.6.3.dist-info/METADATA,sha256=eVjFTr1hxbsZPt6p-7bEYuXvUFpFq2F9El75KWoO4RY,34904
+onnxtr-0.6.3.dist-info/WHEEL,sha256=0CuiUZ_p9E4cD6NyLD6UG80LBXYyiSYZOKDm5lp32xk,91
+onnxtr-0.6.3.dist-info/top_level.txt,sha256=r_MSUTpspp4pWEEWvly-s7ZkfCg1KwrK6-kBlXkWKU8,7
+onnxtr-0.6.3.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+onnxtr-0.6.3.dist-info/RECORD,,

{onnxtr-0.6.0.dist-info → onnxtr-0.6.3.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (75.6.0)
+Generator: setuptools (80.3.1)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

onnxtr-0.6.0.dist-info/RECORD

@@ -1,75 +0,0 @@
-onnxtr/__init__.py,sha256=h7Wc2tuHLsaoCk5xNpEFEK-g11A6SJA7nAasA76TQ_Y,100
-onnxtr/file_utils.py,sha256=LnHd3Ns6TA9AAyYnOnE8hs-obAKYdJfnxo4WshUo300,1067
-onnxtr/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-onnxtr/version.py,sha256=wIe1qRfuQKAuHZHcUFU7RwRIL___wvw0qm3KCxHLyFg,23
-onnxtr/contrib/__init__.py,sha256=0Jo052I1DIdbNVYQBg213UJBNH3UKZIOcn2HVusQlj4,39
-onnxtr/contrib/artefacts.py,sha256=gPy2uD1gHIRsTS3-BSJHPZleqNsWoQkRh465QUJpFJ4,5323
-onnxtr/contrib/base.py,sha256=tA00-Mo4QsJ5Bjtexg2YFnBQ3xIfs3vn-PiMSBXp8Dc,3135
-onnxtr/io/__init__.py,sha256=kS7tKGFvzxOCWBOun-Y8n9CsziwRKNynjwpZEUUI03M,106
-onnxtr/io/elements.py,sha256=k4CLnZiyXt3HvDkIxXsjYqDwNSpksinRKMrr8qCx3Pk,17736
-onnxtr/io/html.py,sha256=fnBj3-wfk1UhRQWgiF6TST54K97N_bxg-XdLtddbk0k,716
-onnxtr/io/image.py,sha256=GBVAC1Au_18Oz6FG4sZKINhspb6-w447WdRYZsKMxEo,1700
-onnxtr/io/pdf.py,sha256=8TD2kFdVRky5-hs8xXgvrM8DspRHaJNVUSg5x4Z3JuI,1327
-onnxtr/io/reader.py,sha256=BAatmgolmAkFtrExtE-sZ1dT8R8rTy8az2RgoKIUlTo,2755
-onnxtr/models/__init__.py,sha256=QTfZlqUyv1d7NUCbGIUFM1DLOOXe-cqHZ7uaKkGdXvk,157
-onnxtr/models/_utils.py,sha256=ep0kXUIUoyvVWQtJBkF7Ma_2NI_-vunMciS5vas_RC8,6424
-onnxtr/models/builder.py,sha256=Q-UdIlQUyT7tAcPiZ5lxsIMrvvvQweYSlTRSzmkg700,14076
-onnxtr/models/engine.py,sha256=q5kAjSMoQNJz0cYD9Td4K6JUfXXeKGjxMaTVjazJR88,4839
-onnxtr/models/zoo.py,sha256=JYwhhU1dKi05dZ_d5Rte5qvTEtEFWqHCmpRHWpGjLJw,5303
-onnxtr/models/classification/__init__.py,sha256=h1bZs55iLJBMATtzS4ntTKwfD6OGXBiiqGv_hEnOFnE,41
-onnxtr/models/classification/zoo.py,sha256=loXyRt0YIbNOUVBTtzioPx0wQTu5nNNmiO8O40w-TZU,4271
-onnxtr/models/classification/models/__init__.py,sha256=rohbM6ZQslfYchi7feZwwh-sX3XXRUhgtEJQeurAytQ,24
-onnxtr/models/classification/models/mobilenet.py,sha256=bAxLLh3WDF2l6LIQMF-iT42EAi_5tQLVh_xV_YF_vXs,4818
-onnxtr/models/classification/predictor/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
-onnxtr/models/classification/predictor/base.py,sha256=Cu-WJalJTmt3jw7kA8yProUqNEtDxyNPsac6RgwkpLs,2313
-onnxtr/models/detection/__init__.py,sha256=h1bZs55iLJBMATtzS4ntTKwfD6OGXBiiqGv_hEnOFnE,41
-onnxtr/models/detection/core.py,sha256=QfJkzhX2RFFPzw6L4YYPK838ych2HcmTLpincSyKyzw,3497
-onnxtr/models/detection/zoo.py,sha256=xoFPyNKFg3gcuYjzRBA54dNkK6NRvVYmy48yp59Kr3M,3324
-onnxtr/models/detection/_utils/__init__.py,sha256=oPkIYbySSbLsOk02wVPNO9bUuywC47YjaenfyTwfOsw,20
-onnxtr/models/detection/_utils/base.py,sha256=2jDJxfmLG1WdBEjFuNXisMQIheLkxv5zZNdBoatpJh8,2300
-onnxtr/models/detection/models/__init__.py,sha256=6Ea6knYrVCR2jAmPlsVWmCdHe-c6lSRETSAuZGfhx8I,85
-onnxtr/models/detection/models/differentiable_binarization.py,sha256=794Ohj6wNJ2aW5iG6EsvF73xH_h7Zm6VaYHycxiD4xY,6630
-onnxtr/models/detection/models/fast.py,sha256=6COxzjFYEy5I1eu6goT7Rg_C4hOzErBFtr2Ewm9_FVM,6188
-onnxtr/models/detection/models/linknet.py,sha256=KBAZfrG_1G2XiSYDzA1oWUeF-YBGAZgM7ICx2e-EaQ4,6666
-onnxtr/models/detection/postprocessor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-onnxtr/models/detection/postprocessor/base.py,sha256=T8uWfjiEamuSX9IHesToz5BeEHUAUijOBf0L9or0FWM,5692
-onnxtr/models/detection/predictor/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
-onnxtr/models/detection/predictor/base.py,sha256=JDA7IzqIYB7QdmWxXl8bXzVA6aCfjtlo0nPlkhlhxu8,2293
-onnxtr/models/factory/__init__.py,sha256=cKPoH2V2157lLMTR2zsljG3_IQHziodqR-XK_LG0D_I,19
-onnxtr/models/factory/hub.py,sha256=voxC3yxElLQYx_dkFLvFIFGZnLJn10KGgJ0NYGL0LBU,7126
-onnxtr/models/predictor/__init__.py,sha256=XL25XkRkgyK7mldF-CWhg2MMakSdP5vLpDLwL59hphk,25
-onnxtr/models/predictor/base.py,sha256=skQ52iKd_Yb06UsHJ4IDlqWjUxeJzxj770ViSUBeki4,9432
-onnxtr/models/predictor/predictor.py,sha256=Ds2k2_5hsq9DCrWoTFunkEvfZnSwKvMp_NZBDRpYiBE,6377
-onnxtr/models/preprocessor/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
-onnxtr/models/preprocessor/base.py,sha256=tGv4FZddcjOXEXKkYF-EqisY7xok13gJ41rKmYxuuGk,3963
-onnxtr/models/recognition/__init__.py,sha256=h1bZs55iLJBMATtzS4ntTKwfD6OGXBiiqGv_hEnOFnE,41
-onnxtr/models/recognition/core.py,sha256=uvlJJcrMkfchdfjXSrxRV3eyGT01e_OTORQ8NbMX1KI,730
-onnxtr/models/recognition/utils.py,sha256=ewxztEeZx7dG8wq6FEwesaAXp_S1mED-h-mnSR4KcbI,3503
-onnxtr/models/recognition/zoo.py,sha256=W0RgI8wde83QLIeSm3k4YHoLzFwmVvCheF2D7I3Jlr4,2857
-onnxtr/models/recognition/models/__init__.py,sha256=IXfiuzzkft8O1CpBZWYTpFw19y49mt5rJ_iGSdaWiU0,105
-onnxtr/models/recognition/models/crnn.py,sha256=1mZrwnxBk_bkSpZwxZw2EZBdRQ_QST4kfElIIKb34ZA,8786
-onnxtr/models/recognition/models/master.py,sha256=nAUiQ5FS7jqSDfOCmgNt3Oh76xPEKMF28VefvXfve7Y,4669
-onnxtr/models/recognition/models/parseq.py,sha256=001zYFYhGv9F_3wneBccHG9KLd8_3ul_dO_OHVnYimI,4512
-onnxtr/models/recognition/models/sar.py,sha256=On-rBYw7eboLDRN8pSBHPaN_XgOUgVoVvZ3ytcMh62k,4523
-onnxtr/models/recognition/models/vitstr.py,sha256=sk2JXxQ0uRqqq0FWpM90MfW8HhHmeQUW8C-ew9lyG4w,5964
-onnxtr/models/recognition/predictor/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
-onnxtr/models/recognition/predictor/_utils.py,sha256=Ee_BfKyhvJJij763VRGTfzqWcft0pGnN_zSSmXAY8fE,3329
-onnxtr/models/recognition/predictor/base.py,sha256=3OREQIsAWPxdKKRyJD5WRAs47AZknff85EeIU5qZsBA,2488
-onnxtr/transforms/__init__.py,sha256=ERmmOxz_9mUkIuccNbzUa5Y6gVLLVDdyc4cCxbCCUbY,20
-onnxtr/transforms/base.py,sha256=Fu0xBf9xdDrbVP9Hc7l3R35TGNANSbRqrFCoxl4OTqo,3887
-onnxtr/utils/__init__.py,sha256=pESRJKtcQyjRxiMgZPhtPYeLbCj-YSGyMVRHTbcMONU,94
-onnxtr/utils/common_types.py,sha256=6fn8g3aLV-cu4QkM_R_Ab8Ihmok16EXvhpuz7Q5no20,551
-onnxtr/utils/data.py,sha256=2nHI3WH5NYZW94LigQwBxG2bSDBXQl9UIu-e3sHq-CQ,4262
-onnxtr/utils/fonts.py,sha256=OnSgUAXn4zkfMuD1AO08zfYyM_qI5FRp39oXP_6kU40,1282
-onnxtr/utils/geometry.py,sha256=sFw66wThJfFKOyaM-kPeA719WHOIBH97mGoLhtE7nhg,20418
-onnxtr/utils/multithreading.py,sha256=4vHWzX__xOnvtlpL_BTQUzUDLhTtJbYFXhm7C_NyIpc,1994
-onnxtr/utils/reconstitution.py,sha256=C2U7qEU2xHBnxN_7bDq3wvCrYG2gf6cWWxWguChKXbY,6119
-onnxtr/utils/repr.py,sha256=CPHCedlVWUMSTjcDpj7YpsV4wu7Z0tXbY1SfCixNvZ0,2105
-onnxtr/utils/visualization.py,sha256=LYMDIIxYlt0B4AxvoGfw463pD9_B88I-811_vwcrSw0,9782
-onnxtr/utils/vocabs.py,sha256=PTqQwd3itXRlX057JV4PjpXlAw_k7Yb5_wxK4CGKFHE,4136
-onnxtr-0.6.0.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-onnxtr-0.6.0.dist-info/METADATA,sha256=8KUWIDZ_gtm9YfH1aq0yf5iitRx5D6PkAMPM4p6k-Fo,34597
-onnxtr-0.6.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-onnxtr-0.6.0.dist-info/top_level.txt,sha256=r_MSUTpspp4pWEEWvly-s7ZkfCg1KwrK6-kBlXkWKU8,7
-onnxtr-0.6.0.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-onnxtr-0.6.0.dist-info/RECORD,,