pytorch 自定义dataset类

发布时间 2023-10-08 14:36:40作者: 心比天高xzh

实现模版

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)