ImageFolder(root, transform=None, target_transform=None, loader=default_loader)
它主要有四个参数:
root:在root指定的路径下寻找图片
transform:对PIL Image进行的转换操作,transform的输入是使用loader读取图片的返回对象
target_transform:对label的转换
loader:给定路径后如何读取图片,默认读取为RGB格式的PIL Image对象
root路径下应将所有文件按照文件夹保存,每个文件夹下存储同一个类别的图片,类别名就是该文件夹的名字。如果某个类别的目录下存在文本文件,并不会抛出异常,因为ImageFolder会维护一部分可接受的图像文件类型,对于不可接受的文件类型会直接忽略掉。
创建实例时,会以类别目录名为第一关键字,类别目录下的图像文件名为第二关键字,按字典序从小到大依次处理,因此ImageFolder维护样本的顺序是确定的。