summaryrefslogtreecommitdiff
path: root/text_recognizer/image_utils.py
diff options
context:
space:
mode:
authorGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-27 23:19:10 +0200
committerGustaf Rydholm <gustaf.rydholm@gmail.com>2022-09-27 23:19:10 +0200
commit49069253b8eba643aecbc1972879be9517801272 (patch)
tree421d9ea20c2db1a9697c4309f692236f32407e8d /text_recognizer/image_utils.py
parentee28ba2ae6fc98d624298938693b52af74f64cf2 (diff)
Move utils
Diffstat (limited to 'text_recognizer/image_utils.py')
-rw-r--r--text_recognizer/image_utils.py34
1 files changed, 0 insertions, 34 deletions
diff --git a/text_recognizer/image_utils.py b/text_recognizer/image_utils.py
deleted file mode 100644
index fcb094c..0000000
--- a/text_recognizer/image_utils.py
+++ /dev/null
@@ -1,34 +0,0 @@
-"""Image util functions for loading and saving images."""
-import base64
-from io import BytesIO
-from pathlib import Path
-from typing import Any, Union
-
-import smart_open
-from PIL import Image
-
-
-def read_image_pil(image_uri: Union[Path, str], grayscale: bool = False) -> Image:
- """Read image from uri."""
- with smart_open(image_uri, "rb") as image_file:
- return read_image_pil_file(image_file, grayscale)
-
-
-def read_image_pil_file(image_file: str, grayscale: bool = False) -> Image:
- """Return PIL image."""
- with Image.open(image_file) as image:
- if grayscale:
- image = image.convert("L")
- else:
- image = image.convert(mode=image.mode)
- return image
-
-
-def read_b64_image(b64_str: str, grayscale: bool = False) -> Image:
- """Load base64-encoded images."""
- try:
- _, b64_data = b64_str.split(",")
- image_file = BytesIO(base64.b64decode(b64_data))
- return read_image_pil_file(image_file, grayscale)
- except Exception as e:
- raise ValueError(f"Could not load image from b64 {b64_str}: {e}")