megengine.data.dataset.ImageFolder

class ImageFolder(root, check_valid_func=None, class_name=False)

基类：megengine.data.dataset.vision.meta_vision.VisionDataset

ImageFolder是一个类，可从有组织的文件夹中加载图像数据和标签。

该文件夹组织结构应为：root/cls/xxx.img_ext

标签（label）值是在根目录中类排序后的索引。

参数

• root (str) – 图像文件夹（image folder）的根目录。

• loader – 从路径中加载图像的函数。如果为``None``，则调用 PIL 加载图像的默认函数。

• check_valid_func – 用来检查文件夹中的文件是否为期望图像文件的函数。如果为 None ，则默认调用检查文件扩展名的函数。

• class_name (bool) – 如果为 True ，那么返回类名称，而不是类索引。

方法

__init__(root[, check_valid_func, class_name])	初始化方法。
collect_class()	返回类型 Dict
collect_samples()	返回类型 List

megengine.data.dataset.VisionDataset megengine.data.dataset.MNIST