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

import torch
from torch.utils.data import Dataset, DataLoader
from torchvision import transforms
from PIL import Image

# 自定义数据集类
class CustomDataset(Dataset):
    def __init__(self, image_paths, transform=None):
        self.image_paths = image_paths
        self.transform = transform

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

    def __getitem__(self, index):
        image_path = self.image_paths[index]
        image = Image.open(image_path).convert('RGB')

        if self.transform:
            image = self.transform(image)

        return image, image_path

# 图片路径列表
image_paths = ['image1.jpg', 'image2.jpg', 'image3.jpg']

# 定义数据预处理的transform
transform = transforms.Compose([
    transforms.Resize((224, 224)),
    transforms.ToTensor()
])

# 创建自定义数据集实例
dataset = CustomDataset(image_paths, transform=transform)

# 创建数据加载器
dataloader = DataLoader(dataset, batch_size=1, shuffle=True)

# 遍历数据加载器
for images, image_paths in dataloader:
    # images为图片数据，image_paths为对应的图片路径
    # 在这里可以进行模型训练或其他操作
    print(images.shape)
    print(image_paths)