Merge pull request #116 from VikParuchuri/dev

Resize change
VikParuchuri · May 28, 2024 · d167369 · d167369
2 parents 31e36e7 + e37597d
commit d167369
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "surya-ocr"
-version = "0.4.9"
+version = "0.4.10"
 description = "OCR, layout, reading order, and line detection in 90+ languages"
 authors = ["Vik Paruchuri <[email protected]>"]
 readme = "README.md"

diff --git a/surya/input/processing.py b/surya/input/processing.py
@@ -48,9 +48,9 @@ def split_image(img, processor):
 def prepare_image_detection(img, processor):
     new_size = (processor.size["width"], processor.size["height"])
 
-    img = np.asarray(img, dtype=np.uint8)
-    img = cv2.resize(img, new_size, interpolation=cv2.INTER_LANCZOS4)
+    img = img.resize(new_size, Image.Resampling.LANCZOS) # Stretch smaller dimension to fit new size
 
+    img = np.asarray(img, dtype=np.uint8)
     img = processor(img)["pixel_values"][0]
     img = torch.from_numpy(img)
     return img