image_load

paddle.vision.image. image_load ( path, backend=None ) [源代码]

读取一个图像。

参数

  • path (str) - 图像路径。

  • backend (str,可选) - 加载图像的后端。参数必须是 cv2pilNone 之一。如果后端为 None,则使用全局的 _imread_backend 参数,默认值为 pil。这个参数可以使用 set_image_backend 指定。默认值:None 。

返回

PIL.Image numpy.ndarray,加载后的图像。

代码示例

>>> import numpy as np
>>> from PIL import Image
>>> from paddle.vision import image_load, set_image_backend

>>> fake_img = Image.fromarray((np.random.random((32, 32, 3)) * 255).astype('uint8'))

>>> path = 'temp.png'
>>> fake_img.save(path)

>>> set_image_backend('pil')

>>> pil_img = image_load(path).convert('RGB')

>>> print(type(pil_img))
<class 'PIL.Image.Image'>

>>> # use opencv as backend
>>> set_image_backend('cv2')

>>> np_img = image_load(path)
>>> print(type(np_img))
<class 'numpy.ndarray'>