11 lines
261 B
Python
11 lines
261 B
Python
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)))
|