Spaces:
Running
Running
| 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 | |