src/cv_utils.py

import cv2
import numpy as np
from PIL import Image as PILImage


def get_image(image_input) -> np.array:
    """Outputs numpy array of image given a string filepath or PIL image"""
    if type(image_input) == str:
        image = cv2.imread(image_input)  # OpenCV uses BGR
    else:
        image = cv2.cvtColor(np.array(image_input), cv2.COLOR_RGB2BGR)  # PIL uses RGB
    return image


def resize_image_height(image: PILImage.Image, new_height=300) -> PILImage.Image:
    aspect_ratio = image.width / image.height
    new_width = int(aspect_ratio * new_height)
    image = image.resize((new_width, new_height))
    return image