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