如何更改pytorch数据文件夹中的标签？

import os
import torch
from torchvision.datasets import ImageFolder

# 定义数据集路径和标签列表
data_folder = 'path/to/dataset'
labels = ['cat', 'dog', 'bird']

# 加载数据集
dataset = ImageFolder(data_folder)

# 遍历数据集中的每个样本
for i in range(len(dataset)):
    # 获取样本的路径和标签
    image_path, label = dataset.samples[i]
    
    # 修改标签
    new_label = labels[label]  # 根据原始标签获取新的标签
    
    # 打印原始标签和新标签
    print(f'Original label: {label}, New label: {new_label}')
    
    # 修改数据集中的标签
    dataset.samples[i] = image_path, new_label

# 保存修改后的数据集
new_data_folder = 'path/to/new_dataset'
os.makedirs(new_data_folder, exist_ok=True)
torch.save(dataset, os.path.join(new_data_folder, 'dataset.pth'))