Dataset、DataLoader、DataLoaderIter

一、Dataset

对单个样本的处理

1、__getitem__：

返回一条数据或一个样本，obj[index] == obj.__getitem__()

2、__len__:

返回整个数据集中样本的数量，len(obj) == obj.__len__()

二、DataLoader

对batch个样本的处理

主要完成两件事：

a、定义了一堆成员变量，后面赋值给DataLoaderIter

b、将自身传入DataLoaderIter

三、DataLoaderIter

Dataset装入DataLoader，DataLoader装入DataLoaderIter

for data in DataLoader:

a、调用DataLoader的__iter__()得到迭代器DataLoaderIter

b、循环调用DataLoaderIter的__next__函数

原文：https://www.cnblogs.com/liujianing/p/13658261.html