pytorch 自定义dataset类

实现模版

class our_dataset(Dataset):
    def __init__(self,···):
        super(our_dataset, self).__init__()
        #初始化，可以自定义添加参数
    def __getitem__(self,index):
        ···
        return img,label
        #根据索引(0,len(dataset)-1)获取样本对，包括输入与输出
    def __len__(self):
        return len(dataset)
        #表示数据集的长度，len(dataset)