Question

我下载了一个数据集，其中包含根据目录中标签排列的ASL字母图像。每个目录的名称是其中包含图像的标签。现在，当我使用torch.datasets.ImageFolder()时，它可以很好地加载数据集，但是它分配的标签不是文件夹的名称。相反，它只是分配一个数字作为标签（可能按文件夹名称的字母顺序）。

def loadData(self, PATH):
    transform = transforms.Compose([transforms.ToTensor()])
    dataset = datasets.ImageFolder(root = PATH, transform = transform)

我尝试使用dataset.targets[]方法，但无法正常工作

def loadData(self, PATH):
    transform = transforms.Compose([transforms.ToTensor()])
    dataset = datasets.ImageFolder(root = PATH, transform = transform)
    for i in range(26):
        dataset.targets[i] = chr(i+65)
    dataset.targets[26]= 'SPACE'
    dataset.targets[27]= 'DELETE'
    dataset.targets[28]= 'NOTHING'

这是我打印一些随机标签时得到的输出：

tensor(5)
tensor(1)
tensor(7)
tensor(6)
tensor(24)
tensor(12)

如何在pytorch中更改自定义数据集的标签？

0 个答案: