LEv145
764c5e54f9
Modular utils Typo `create_image_grid_by_columns` -> `create_images_grid_by_columns` Typo `create_image_grid_by_rows` -> `create_images_grid_by_rows`
14 lines
364 B
Python
14 lines
364 B
Python
import typing as t
|
|
|
|
import torch
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
|
|
def tensor_to_pillow(image: t.Any) -> Image.Image:
|
|
return Image.fromarray(np.clip(255. * image.cpu().numpy().squeeze(), 0, 255).astype(np.uint8))
|
|
|
|
|
|
def pillow_to_tensor(image: Image.Image) -> t.Any:
|
|
return torch.from_numpy(np.array(image).astype(np.float32) / 255.0).unsqueeze(0)
|