diff options
author | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-02-06 22:46:01 +0100 |
---|---|---|
committer | Gustaf Rydholm <gustaf.rydholm@gmail.com> | 2022-02-06 22:46:01 +0100 |
commit | 8103ab92b9d8f1898164ded197d7ec60671e2aca (patch) | |
tree | ade6d8b833ca77c3ae1b795b5abbd7e1350fcfa1 /text_recognizer/data/utils | |
parent | 2d4e527fc0d47db8ed995180cbcec01271e20245 (diff) |
fix: refactor and move image utils
Diffstat (limited to 'text_recognizer/data/utils')
-rw-r--r-- | text_recognizer/data/utils/image_utils.py | 49 |
1 files changed, 0 insertions, 49 deletions
diff --git a/text_recognizer/data/utils/image_utils.py b/text_recognizer/data/utils/image_utils.py deleted file mode 100644 index c2b8915..0000000 --- a/text_recognizer/data/utils/image_utils.py +++ /dev/null @@ -1,49 +0,0 @@ -"""Image util functions for loading and saving images.""" -from pathlib import Path -from typing import Union -from urllib.request import urlopen - -import cv2 -import numpy as np -from PIL import Image - - -def read_image_pil(image_uri: Union[Path, str], grayscale: bool = False) -> Image: - """Return PIL image.""" - image = Image.open(image_uri) - if grayscale: - image = image.convert("L") - return image - - -def read_image(image_uri: Union[Path, str], grayscale: bool = False) -> np.array: - """Read image_uri.""" - - if isinstance(image_uri, str): - image_uri = Path(image_uri) - - def read_image_from_filename(image_filename: Path, imread_flag: int) -> np.array: - return cv2.imread(str(image_filename), imread_flag) - - def read_image_from_url(image_url: Path, imread_flag: int) -> np.array: - url_response = urlopen(str(image_url)) # nosec - image_array = np.array(bytearray(url_response.read()), dtype=np.uint8) - return cv2.imdecode(image_array, imread_flag) - - imread_flag = cv2.IMREAD_GRAYSCALE if grayscale else cv2.IMREAD_COLOR - image = None - - if image_uri.exists(): - image = read_image_from_filename(image_uri, imread_flag) - else: - image = read_image_from_url(image_uri, imread_flag) - - if image is None: - raise ValueError(f"Could not load image at {image_uri}") - - return image - - -def write_image(image: np.ndarray, filename: Union[Path, str]) -> None: - """Write image to file.""" - cv2.imwrite(str(filename), image) |