onnxtr 0.6.2__tar.gz → 0.6.3__tar.gz

{onnxtr-0.6.2 → onnxtr-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: onnxtr
-Version: 0.6.2
+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,7 +292,8 @@ 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.2-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)

{onnxtr-0.6.2 → onnxtr-0.6.3}/README.md

@@ -7,7 +7,8 @@
 [![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.2-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)

{onnxtr-0.6.2 → onnxtr-0.6.3}/onnxtr/io/elements.py

@@ -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-0.6.2 → onnxtr-0.6.3}/onnxtr/models/builder.py

@@ -74,7 +74,7 @@ class DocumentBuilder(NestedObject):
         """
         lines = []
         # Sort words horizontally
-        word_idcs = [word_idcs[idx] for idx in boxes[word_idcs, 0].argsort().tolist()]  # type: ignore[call-overload]
+        word_idcs = [word_idcs[idx] for idx in boxes[word_idcs, 0].argsort().tolist()]
 
         # Eventually split line horizontally
         if len(word_idcs) < 2:

{onnxtr-0.6.2 → onnxtr-0.6.3}/onnxtr/models/detection/postprocessor/base.py

@@ -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-0.6.2 → onnxtr-0.6.3}/onnxtr/models/engine.py

@@ -4,6 +4,7 @@
 # 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

{onnxtr-0.6.2 → onnxtr-0.6.3}/onnxtr/models/recognition/models/master.py

@@ -106,7 +106,7 @@ class MASTERPostProcessor(RecognitionPostProcessor):
             "".join(self._embedding[idx] for idx in encoded_seq).split("<eos>")[0] for encoded_seq in out_idxs
         ]
 
-        return list(zip(word_values, np.clip(probs, 0, 1).astype(float).tolist()))  # type: ignore[arg-type]
+        return list(zip(word_values, np.clip(probs, 0, 1).astype(float).tolist()))
 
 
 def _master(

{onnxtr-0.6.2 → onnxtr-0.6.3}/onnxtr/transforms/base.py

@@ -4,6 +4,8 @@
 # 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-0.6.2 → onnxtr-0.6.3}/onnxtr/utils/visualization.py

@@ -274,11 +274,11 @@ def draw_boxes(boxes: np.ndarray, image: np.ndarray, color: tuple[int, int, int]
     _boxes[:, [1, 3]] *= h
     _boxes = _boxes.astype(np.int32)
     for box in _boxes.tolist():
-        xmin, ymin, xmax, ymax = box  # type: ignore[misc]
+        xmin, ymin, xmax, ymax = box
         image = cv2.rectangle(
             image,
-            (xmin, ymin),  # type: ignore[arg-type]
-            (xmax, ymax),  # type: ignore[arg-type]
+            (xmin, ymin),
+            (xmax, ymax),
             color=color if isinstance(color, tuple) else (0, 0, 255),
             thickness=2,
         )

{onnxtr-0.6.2 → onnxtr-0.6.3}/onnxtr/utils/vocabs.py

@@ -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"]
-    + "áàảạãăắằẳẵặâấầẩẫậđéèẻẽẹêếềểễệóòỏõọôốồổộỗơớờởợỡúùủũụưứừửữựíìỉĩịýỳỷỹỵ"
-    + "ÁÀẢẠÃĂẮẰẲẴẶÂẤẦẨẪẬĐÉÈẺẼẸÊẾỀỂỄỆÓÒỎÕỌÔỐỒỔỘỖƠỚỜỞỢỠÚÙỦŨỤƯỨỪỬỮỰÍÌỈĨỊÝỲỶỸỴ"
+
+# 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-0.6.3/onnxtr/version.py

@@ -0,0 +1 @@
+__version__ = 'v0.6.3'

{onnxtr-0.6.2 → onnxtr-0.6.3}/onnxtr.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: onnxtr
-Version: 0.6.2
+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,7 +292,8 @@ 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.2-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)

{onnxtr-0.6.2 → onnxtr-0.6.3}/onnxtr.egg-info/requires.txt

@@ -2,7 +2,6 @@ numpy<3.0.0,>=1.16.0
 scipy<2.0.0,>=1.4.0
 pypdfium2<5.0.0,>=4.11.0
 pyclipper<2.0.0,>=1.2.0
-shapely<3.0.0,>=1.6.0
 rapidfuzz<4.0.0,>=3.0.0
 langdetect<2.0.0,>=1.0.9
 huggingface-hub<1.0.0,>=0.23.0
@@ -28,6 +27,7 @@ mplcursors>=0.3
 pytest>=5.3.2
 coverage[toml]>=4.5.4
 requests>=2.20.0
+pytest-memray>=1.7.0
 ruff>=0.1.5
 mypy>=0.812
 pre-commit>=2.17.0
@@ -60,6 +60,7 @@ pre-commit>=2.17.0
 pytest>=5.3.2
 coverage[toml]>=4.5.4
 requests>=2.20.0
+pytest-memray>=1.7.0
 
 [viz]
 matplotlib>=3.1.0

{onnxtr-0.6.2 → onnxtr-0.6.3}/pyproject.toml

@@ -35,7 +35,6 @@ dependencies = [
     "scipy>=1.4.0,<2.0.0",
     "pypdfium2>=4.11.0,<5.0.0",
     "pyclipper>=1.2.0,<2.0.0",
-    "shapely>=1.6.0,<3.0.0",
     "rapidfuzz>=3.0.0,<4.0.0",
     "langdetect>=1.0.9,<2.0.0",
     "huggingface-hub>=0.23.0,<1.0.0",
@@ -81,6 +80,7 @@ testing = [
     "pytest>=5.3.2",
     "coverage[toml]>=4.5.4",
     "requests>=2.20.0",
+    "pytest-memray>=1.7.0",
 ]
 quality = [
     "ruff>=0.1.5",
@@ -100,6 +100,7 @@ dev = [
     "pytest>=5.3.2",
     "coverage[toml]>=4.5.4",
     "requests>=2.20.0",
+    "pytest-memray>=1.7.0",
     # Quality
     "ruff>=0.1.5",
     "mypy>=0.812",
@@ -139,7 +140,6 @@ module = [
 	"matplotlib.*",
     "numpy.*",
 	"pyclipper.*",
-	"shapely.*",
 	"mplcursors.*",
 	"defusedxml.*",
 	"weasyprint.*",

{onnxtr-0.6.2 → onnxtr-0.6.3}/setup.py

@@ -9,7 +9,7 @@ from pathlib import Path
 from setuptools import setup
 
 PKG_NAME = "onnxtr"
-VERSION = os.getenv("BUILD_VERSION", "0.6.2a0")
+VERSION = os.getenv("BUILD_VERSION", "0.6.3a0")
 
 
 if __name__ == "__main__":

onnxtr-0.6.2/onnxtr/version.py

@@ -1 +0,0 @@
-__version__ = 'v0.6.2'