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 ,那么返回类名称,而不是类索引。