TypeError：“元组”对象在创建数据生成器 PyTorch 时不能解释为整数

Question

我是 PyTorch 的新手，我正在学习创建批量数据以进行分割。代码如下：

class NumbersDataset(Dataset):
    def __init__(self):
        self.X = list(df['input_img'])
        self.y = list(df['mask_img'])

    def __len__(self):
        return len(self.X), len(self.y)

    def __getitem__(self, idx):
        return self.X[idx], self.y[idx]


if __name__ == '__main__':
    dataset = NumbersDataset()
    dataloader = DataLoader(dataset, batch_size=50, shuffle=True, num_workers=2)
    # print(len(dataset))
    # plt.imshow(dataset[100])
    # plt.show()
    print(next(iter(dataloader)))

其中df['input_img'] 列包含图像的位置 ('/path/to/pic/480p/boxing-fisheye/00010.jpg')，df['mask_img'] 包含所有掩码图像的位置。我正在尝试加载图像，但出现错误：

TypeError: 'tuple' 对象不能被解释为整数

但是，如果我不使用 DataLoader 并执行以下操作：

     dataset = NumbersDataset()
     print(len(dataset))
     print(dataset[10:20])

然后我得到了我所期望的。谁能告诉我我做错了什么？

Answer 1

您不能为 __len__ 方法返回 tuple。预期类型为int

# perhaps you can add the list length's for the total length
# but no matter how you choose to implement the method you can
# only return on value of type integer `int`
def __len__(self):
    return len(self.X) + len(self.y)