| from typing import Tuple, Optional | |
| import numpy as np | |
| from PIL import Image | |
| from imgutils.data import rgb_encode | |
| def _img_encode(image: Image.Image, size: Tuple[int, int] = (384, 384), | |
| normalize: Optional[Tuple[float, float]] = (0.5, 0.5)): | |
| image = image.resize(size, Image.BILINEAR) | |
| data = rgb_encode(image, order_='CHW') | |
| if normalize is not None: | |
| mean_, std_ = normalize | |
| mean = np.asarray([mean_]).reshape((-1, 1, 1)) | |
| std = np.asarray([std_]).reshape((-1, 1, 1)) | |
| data = (data - mean) / std | |
| return data.astype(np.float32) | |