From 46a1472d33d3a4180798492e819f2ec02bc3b1a3 Mon Sep 17 00:00:00 2001 From: Gustaf Rydholm Date: Sun, 28 Mar 2021 22:02:24 +0200 Subject: Add refactor of iam lines --- text_recognizer/util.py | 52 ------------------------------------------------- 1 file changed, 52 deletions(-) delete mode 100644 text_recognizer/util.py (limited to 'text_recognizer/util.py') diff --git a/text_recognizer/util.py b/text_recognizer/util.py deleted file mode 100644 index b431e22..0000000 --- a/text_recognizer/util.py +++ /dev/null @@ -1,52 +0,0 @@ -"""Utility functions for text_recognizer module.""" -import os -from pathlib import Path -from typing import Union -from urllib.request import urlopen - -import cv2 -import numpy as np - - -def read_image(image_uri: Union[Path, str], grayscale: bool = False) -> np.ndarray: - """Read image_uri.""" - - def read_image_from_filename(image_filename: str, imread_flag: int) -> np.ndarray: - return cv2.imread(str(image_filename), imread_flag) - - def read_image_from_url(image_url: str, imread_flag: int) -> np.ndarray: - if image_url.lower().startswith("http"): - url_response = urlopen(str(image_url)) - image_array = np.array(bytearray(url_response.read()), dtype=np.uint8) - return cv2.imdecode(image_array, imread_flag) - else: - raise ValueError( - "Url does not start with http, therefore not safe to open..." - ) from None - - imread_flag = cv2.IMREAD_GRAYSCALE if grayscale else cv2.IMREAD_COLOR - local_file = os.path.exists(image_uri) - image = None - - if local_file: - 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 rescale_image(image: np.ndarray) -> np.ndarray: - """Rescale image from [0, 1] to [0, 255].""" - if image.max() <= 1.0: - image = 255 * (image - image.min()) / (image.max() - image.min()) - return image - - -def write_image(image: np.ndarray, filename: Union[Path, str]) -> None: - """Write image to file.""" - image = rescale_image(image) - cv2.imwrite(str(filename), image) -- cgit v1.2.3-70-g09d2