load_image

torchtune.data.load_image(image_loc: Union[Path, str]) → PIL.Image.Image

从本地文件路径或远程源以 PIL 格式加载图像的便捷方法。

参数:

image_loc (Union[Path, str]) – 指向将以 PIL 格式加载的图像的本地文件路径或远程源。

注意

如果从远程源加载图像，则该函数期望在 image_loc 中提供的 URL 以 “http” 或 “https” 开头，例如 “https://www.wikipedia.org/en/bird.jpg”。

引发:

• ValueError – 如果无法从远程源加载图像。

• ValueError – 如果无法将图像打开为 Image。

示例

>>> # Load from remote source
>>> image = load_image("https://www.wikipedia.org/en/bird.jpg")

>>> # Load from local file path
>>> image = load_image(Path("/home/user/bird.jpg"))

返回:

加载的图像。

返回类型:

PIL.Image.Image