pytorch重写Dataset
对于这个类的重写,需要修改三个魔法方法,分别是__init__
,__getitem__
,__len__
__init__
:
构造函数,接收路径和标签,然后整理成两个列表,以便于在__getitem__
中进行索引提取。
__len__
:
直接返回样本个数即可。
__getitem__
:
这个函数的作用是收取一个索引,然后返回对应的文件路径和标签值。这个路径和标签值已经在__init__
中定义好了。
不过有的模型比较复杂,需要返回多个元素,这时候可以使用字典,return一个字典。
例如:
sample = {'mri': mri, 'clinical':clinical, 'label':label}
return sample