o QeÓã@s>ddlmZddlmZgZdadd„Zdd„Zd d d „ZdS) é)ÚImage)Ú try_importÚpilcCs|dvr td |¡ƒ‚|adS)aò Specifies the backend used to load images in class ``paddle.vision.datasets.ImageFolder`` and ``paddle.vision.datasets.DatasetFolder`` . Now support backends are pillow and opencv. If backend not set, will use 'pil' as default. Args: backend (str): Name of the image load backend, should be one of {'pil', 'cv2'}. Examples: .. code-block:: python import os import shutil import tempfile import numpy as np from PIL import Image from paddle.vision import DatasetFolder from paddle.vision import set_image_backend set_image_backend('pil') def make_fake_dir(): data_dir = tempfile.mkdtemp() for i in range(2): sub_dir = os.path.join(data_dir, 'class_' + str(i)) if not os.path.exists(sub_dir): os.makedirs(sub_dir) for j in range(2): fake_img = Image.fromarray((np.random.random((32, 32, 3)) * 255).astype('uint8')) fake_img.save(os.path.join(sub_dir, str(j) + '.png')) return data_dir temp_dir = make_fake_dir() pil_data_folder = DatasetFolder(temp_dir) for items in pil_data_folder: break # should get PIL.Image.Image print(type(items[0])) # use opencv as backend # set_image_backend('cv2') # cv2_data_folder = DatasetFolder(temp_dir) # for items in cv2_data_folder: # break # should get numpy.ndarray # print(type(items[0])) shutil.rmtree(temp_dir) ©rÚcv2Ztensorú@Expected backend are one of ['pil', 'cv2', 'tensor'], but got {}N)Ú ValueErrorÚformatÚ_image_backend)Úbackend©r úCD:\Projects\ConvertPro\env\Lib\site-packages\paddle/vision/image.pyÚset_image_backends <þrcCstS)a Gets the name of the package used to load images Returns: str: backend of image load. Examples: .. code-block:: python from paddle.vision import get_image_backend backend = get_image_backend() print(backend) )r r r r r Úget_image_backendZsrNcCsR|durt}|dvrtd |¡ƒ‚|dkrt |¡S|dkr'tdƒ}| |¡SdS)aILoad an image. Args: path (str): Path of the image. backend (str, optional): The image decoding backend type. Options are `cv2`, `pil`, `None`. If backend is None, the global _imread_backend specified by ``paddle.vision.set_image_backend`` will be used. Default: None. Returns: PIL.Image or np.array: Loaded image. Examples: .. code-block:: python 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)) Nrrrr)r rr rÚopenrZimread)Úpathr rr r r Ú image_loadns)þ  þr)N) ZPILrZ paddle.utilsrÚ__all__r rrrr r r r Ús  C