Как изменить метки настраиваемого набора данных в pytorch?

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)