PyTorch DataLoader
torch.utils.data.Dataset是一个抽象类,表示一个数据集。自定义的数据集要继承这个类,并且重写__len__和__getitem__方法。
__len__:len(dataset)返回数据集的size。
__getitem__:支持索引,dataset[i]用来获取第i个样本。
在__init__方法中读csv标签文件,在__getitem__方法中读图片,这样可以按需读取,而不用一次性把所有图片都存在内存中。
参考链接
https://pytorch.org/tutorials/beginner/data_loading_tutorial.html
https://discuss.pytorch.org/t/guidelines-for-assigning-num-workers-to-dataloader/813/24