megengine.data.dataset.ImageFolder

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

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

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

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

参数

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

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

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

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

方法

collect_class()	rtype Dict
collect_samples()	rtype List

上一页

megengine.data.dataset.VisionDataset

下一页

megengine.data.dataset.ImageFolder.collect_class