pytorch：如何在dataloader中使用图片作为标签？

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

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

---

提前谢谢

编写自定义的

数据集

，下面是一个简单的示例

将torch.utils.data.Dataset作为数据集导入
类CustomDataset（数据集）：
定义初始化（自我、输入、标签、转换）：
self.input\u imgs=input\u imgs
self.label\u imgs=label\u imgs
self.transform=transform
定义（自我）：
返回长度（自输入\u imgs）
def uu getitem uu（self，idx）：
input\u img，label\u img=self.input\u imgs[idx]，self.label\u imgs[idx]
返回self.transform（输入），self.transform（标签）

然后，将其传递给Dataloader：

dataloader=dataloader（CustomDataset）

编写自定义的

数据集

，下面是一个简单的示例

将torch.utils.data.Dataset作为数据集导入
类CustomDataset（数据集）：
定义初始化（自我、输入、标签、转换）：
self.input\u imgs=input\u imgs
self.label\u imgs=label\u imgs
self.transform=transform
定义（自我）：
返回长度（自输入\u imgs）
def uu getitem uu（self，idx）：
input\u img，label\u img=self.input\u imgs[idx]，self.label\u imgs[idx]
返回self.transform（输入），self.transform（标签）

然后，将其传递给Dataloader：

dataloader=dataloader（CustomDataset）

我想知道我的数据是否组织为train/input/001.png…002.png train/label/001.png，我应该如何使用上面的代码？我已经投票了，但我的声誉低于15，所以不会公开显示。我想知道我的数据是否组织为train/input/001.png…002.png train/label/001.png，我应该如何使用上面的代码？我已经投票了，但我的声誉低于15，所以不会公开展示。