John6666's picture
Upload 57 files
184d241 verified
raw
history blame contribute delete
537 Bytes
import cv2
import numpy as np
MAX_IMAGE_SIZE = 512
def resize_image(input_image, resolution=MAX_IMAGE_SIZE, interpolation=None):
H, W, C = input_image.shape
H = float(H)
W = float(W)
k = float(resolution) / max(H, W)
H *= k
W *= k
H = int(np.round(H / 64.0)) * 64
W = int(np.round(W / 64.0)) * 64
if interpolation is None:
interpolation = cv2.INTER_LANCZOS4 if k > 1 else cv2.INTER_AREA
img = cv2.resize(input_image, (W, H), interpolation=interpolation)
return img