image_load¶
读取一个图像。
参数¶
path (str) - 图像路径。
backend (str,可选) - 加载图像的后端。参数必须是
cv2
,pil
,None
之一。如果后端为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'>