from PIL import Image import numpy as np def resize_image(image, max_size=1024): w, h = image.size if max(w, h) > max_size: scale = max_size / max(w, h) return image.resize((int(w * scale), int(h * scale))) return image