Spaces:
Running
Running
File size: 632 Bytes
e5ce3a7 0347340 e5ce3a7 0347340 988dde3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
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
|