Dataset
torch.utils.data.Dataset 代表该类数据的抽象类,可以自己定义数据类继承和重写这个抽象类,只需要定义__len__和__getitem__两个函数
DataLoader
通过上述可以定义需要的数据类,通过迭代取得每一个数据,但其中很难取batch、shuffle等操作,需要通过torch.utils.data.DataLoader定义新的迭代器
ImageFloader
关于计算机视觉的还有一个数据读取类ImageFolder,主要用于处理图片
要求图片存放形式:
调用该类:
root表示根目录,在这个目录下有几个文件夹,每个文件夹表示一类
transform表示图像增强,可以单独设置
loader是图片读取的方法