import numpy as np from PIL.Image import Image def image_to_normalized_tensor(image: Image) -> np.ndarray: return np.array(image) / 255 def images_to_batch_tensor(images) -> np.ndarray: return np.array(list(map(image_to_normalized_tensor, images)))