pytorch重写Dataset

pytorch重写Dataset

对于这个类的重写，需要修改三个魔法方法，分别是__init__,__getitem__,__len__

__init__：
构造函数，接收路径和标签，然后整理成两个列表，以便于在__getitem__中进行索引提取。

__len__：
直接返回样本个数即可。
__getitem__:
这个函数的作用是收取一个索引，然后返回对应的文件路径和标签值。这个路径和标签值已经在__init__中定义好了。

不过有的模型比较复杂，需要返回多个元素，这时候可以使用字典，return一个字典。
例如：

sample = {'mri': mri, 'clinical':clinical, 'label':label}
return sample