File size: 513 Bytes
b91f38a | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 | from PIL import Image
def get_imgs_from_pickle(img, transform=None):
"""
将 H5/PKL 里读出的 numpy 图像数组转为 RGB PIL，再按需应用 transform。
参数:
img: numpy.ndarray，形状一般为 (H, W) 或 (H, W, C)
transform: 可选的 torchvision transforms
返回:
已转换的图像（通常为张量）
"""
pil_img = Image.fromarray(img).convert("RGB")
if transform is not None:
pil_img = transform(pil_img)
return pil_img
|