Pytorch中加载数据模块学习

主要函数有Dataset和Dataloader两部分：

        Dataset主要是获取数据，并提取对应的label.(它可以输出数据、并对数据进行编号（以后可以调用数据），还可以获得数据的label)

        Dataloader主要是对数据进行打包，为网络提供不同的数据形式。

1、如何使用Dataset:

Dataset是一个抽象类，所有的数据集都应该继承该类，所有子类都需要重写"__getitem__",获取每一个数据及其对应的label。

初始化类：根据该类创建一个特例、实例需要运行的函数。

def __getitem(self, item):
    pass

item是所有图片的地址

所有图片的地址需要放入到列表中：

img_path_list = os.listdir(path)                   # 该路径下的所有图片的地址，皆存入列表
img_path_list[0]是第一个图片的名称