.. py:class:: paddle.vision.datasets.ImageFolder(root, loader=None, extensions=None, transform=None, is_valid_file=None)
一种通用的数据加载方式,数据需要以如下的格式存放:
root/1.ext
root/2.ext
root/sub_dir/3.ext
- root (str) - 根目录路径。
- loader (Callable,可选) - 可以加载数据路径的一个函数,如果该值没有设定,默认使用
cv2.imread
。默认值为 None。- extensions (list[str]|tuple[str],可选) - 允许的数据后缀列表,
extensions
和is_valid_file
不可以同时设置。如果该值没有设定,默认为('.jpg', '.jpeg', '.png', '.ppm', '.bmp', '.pgm', '.tif', '.tiff', '.webp')
。默认值为 None。- transform (Callable,可选) - 图片数据的预处理,若为
None
即为不做预处理。默认值为None
。- is_valid_file (Callable,可选) - 根据每条数据的路径来判断是否合法的一个函数。
extensions
和is_valid_file
不可以同时设置。默认值为 None。
:ref:`cn_api_io_cn_Dataset`,ImageFolder 实例。
- samples (list[str]) - 样本路径列表。
COPY-FROM: paddle.vision.datasets.ImageFolder