pytorch自定义dataset的两种方法

Pytorch自定义加载数据--自定义Dataset_xuan_liu123的博客-CSDN博客_dataset pytorch

总结：

方法一：继承Dataset类，自定义一个Dataset的数据集类，并改写__init__, __len__,__getitme__函数。

方法二：使用torchvision.datasets.imageFolder函数生成数据集。

注意：原博客中的方法一种，有两个错误：

1.改写__init__函数时，需要添加对父类的初始化：

super(AnimalData, self)__init__()

2.使用transoform转换图像时，输入应该是image，而不是sample字典。