image_load¶
读取一个图像。
参数¶
path (str) - 图像路径。
backend (str,可选) - 加载图像的后端。参数必须是
cv2
,pil
,None
之一。如果后端为None
,则使用全局的_imread_backend
参数,默认值为pil
。这个参数可以使用paddle.vision.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')
# should be PIL.Image.Image
print(type(pil_img))
# use opencv as backend
# set_image_backend('cv2')
# np_img = image_load(path)
# # should get numpy.ndarray
# print(type(np_img))