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