torch.utils.data.Dataset是一个抽象类，表示一个数据集。自定义的数据集要继承这个类，并且重写__len__和__getitem__方法。
__len__：len(dataset)返回数据集的size。
__getitem__：支持索引，dataset[i]用来获取第i个样本。
在__init__方法中读csv标签文件，在__getitem__方法中读图片，这样可以按需读取，而不用一次性把所有图片都存在内存中。
参考链接
本文共 278 字，大约阅读时间需要 1 分钟。
torch.utils.data.Dataset是一个抽象类，表示一个数据集。自定义的数据集要继承这个类，并且重写__len__和__getitem__方法。
__len__：len(dataset)返回数据集的size。
__getitem__：支持索引，dataset[i]用来获取第i个样本。
在__init__方法中读csv标签文件，在__getitem__方法中读图片，这样可以按需读取，而不用一次性把所有图片都存在内存中。
参考链接
转载于:https://www.cnblogs.com/yangwenhuan/p/11381535.html