set_image_backend

paddle.vision.set_image_backend ( backend ) [源代码]

在 paddle.vision.datasets.ImageFolder 和 paddle.vision.datasets.DatasetFolder 类中指定用于加载图像的后端。 现在支持后端是 pillow 和 opencv 。 如果未设置后端，则默认使用 pil 。

参数

  • backend (str) - 加载图像的后端，必须为 pil 或者 cv2 。

代码示例

  1. import os
  2. import shutil
  3. import tempfile
  4. import numpy as np
  5. from PIL import Image
  7. from paddle.vision import DatasetFolder
  8. from paddle.vision import set_image_backend
  10. set_image_backend('pil')
  12. def make_fake_dir():
  13.     data_dir = tempfile.mkdtemp()
  15.     for i in range(2):
  16.         sub_dir = os.path.join(data_dir, 'class_' + str(i))
  17.         if not os.path.exists(sub_dir):
  18.             os.makedirs(sub_dir)
  19.         for j in range(2):
  20.             fake_img = Image.fromarray((np.random.random((32, 32, 3)) * 255).astype('uint8'))
  21.             fake_img.save(os.path.join(sub_dir, str(j) + '.png'))
  22.     return data_dir
  24. temp_dir = make_fake_dir()
  26. pil_data_folder = DatasetFolder(temp_dir)
  28. for items in pil_data_folder:
  29.     break
  31. # should get PIL.Image.Image
  32. print(type(items[0]))
  34. # use opencv as backend
  35. # set_image_backend('cv2')
  37. # cv2_data_folder = DatasetFolder(temp_dir)
  39. # for items in cv2_data_folder:
  40. #     break
  42. # should get numpy.ndarray
  43. # print(type(items[0]))
  45. shutil.rmtree(temp_dir)