Spaces:
Sleeping
Sleeping
File size: 743 Bytes
94a0cd2 2d42726 94a0cd2 2d42726 94a0cd2 2d42726 94a0cd2 2d42726 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
from typing import Tuple
from PIL import Image
from torchvision.transforms import ToTensor
to_tensor = ToTensor()
def preprocess_image(
image: Image, resize_shape: Tuple[int, int] = (256, 256), center_crop=True
):
pil_image = image
if center_crop:
width, height = image.size
crop_size = min(width, height)
left = (width - crop_size) // 2
top = (height - crop_size) // 2
right = (width + crop_size) // 2
bottom = (height + crop_size) // 2
pil_image = image.crop((left, top, right, bottom))
pil_image = pil_image.resize(resize_shape)
tensor_image = to_tensor(pil_image)
tensor_image = tensor_image.unsqueeze(0) * 2 - 1
return pil_image, tensor_image |