pytorch：如何在数据加载器中使用图片作为标签？

Question

我想在pytorch中使用自动编码器进行一些图像重建，但是，我没有找到使用图像作为输入图像标签的方法。（标签图像与原始图像不同）

我试过图像文件夹方法，但我认为这是为了分类，我目前无法想出一个解决方案。我应该为此创建一个自定义数据集吗...

提前致谢！

Answer 1

写下你的习惯Dataset，下面是一个简单的例子。


import torch.utils.data.Dataset as Dataset

class CustomDataset(Dataset):

    def __init__(self, input_imgs, label_imgs, transform):

        self.input_imgs = input_imgs
        self.label_imgs = label_imgs
        self.transform = transform

    def __len__(self):
        return len(self.input_imgs)


    def __getitem__(self, idx):
        input_img, label_img = self.input_imgs[idx], self.label_imgs[idx]
        return self.transform(input_img), self.transform(label_img)

然后，将其传递给 Dataloader：

dataloader = DataLoader(CustomDataset)

pytorch：如何在数据加载器中使用图片作为标签？

pytorch: how can I use picture as label in dataloader?

autoencoder

pytorch