Pytorch框架-Datasets&DataLoaders

Pytorch教程
参考链接：https://github.com/yunjey/pytorch-tutorial
https://pytorch.org/tutorials/beginner/basics/saveloadrun_tutorial.html

1 Datasets & DataLoaders

• 提供两数据加载函数：torch.utils.data.DataLoader和torch.utils.data.Dataset
• 实现数据集代码与模型训练代码分离，以获得更好的可读性和模块化
• 制作自己的数据集必须要实现三个函数：

__init__, __len__,  __getitem__

• init函数在实例化Dataset对象时运行一次
• len返回数据集中样本的数量
• getitem函数的作用是:从给定索引idx,从数据集中加载并返回一个样本并将其转换为张量。

2 文夹结构

在Code 下面新建pytorch_test文件夹，
下面分别建立
data,数据文件夹
datasets,数据集处理文件
tset_dataloader.py成测试文件

path = ‘./pytorch_test/data/dtu/scan1/images’