dataloader读取文件之后会有变量存储文件名称吗

在 PyTorch 中使用 `torch.utils.data.DataLoader` 加载数据时，通常会通过传递一个 `Dataset` 对象来指定数据集，并且在 `Dataset` 中会存储每个样本的数据以及对应的标签等信息。但是，`DataLoader` 并不会自动将文件名存储到变量中。

如果需要将文件名存储到变量中，可以在 `Dataset` 中自定义一个方法，例如 `get_filename(self, index)`，用于获取指定索引下的文件名。在该方法中，可以通过 `self.images[index]` 或者 `self.labels[index]` 等方式获取相应的数据和标签，并通过 `os.path.basename` 方法获取文件名。然后，在训练过程中可以将文件名存储到一个列表或者字典中，以便后续使用。

以下是一个示例代码：

import os
from torch.utils.data import Dataset

class MyDataset(Dataset):
    def __init__(self, data_dir):
        self.data_dir = data_dir
        self.images = [...]  # 存储所有图片的数据
        self.labels = [...]  # 存储所有图片的标签

    def __getitem__(self, index):
        # 获取指定索引下的数据和标签，并返回
        image = self.images[index]
        label = self.labels[index]
        return image, label

    def __len__(self):
        # 返回数据集中的样本数量
        return len(self.images)

    def get_filename(self, index):
        # 获取指定索引下的文件名，并返回
        image_path = os.path.join(self.data_dir, str(index) + '.jpg')
        filename = os.path.basename(image_path)
        return filename

在训练过程中，可以通过以下方式获取文件名：

dataset = MyDataset(data_dir)
dataloader = DataLoader(dataset, batch_size=32, shuffle=True)
filenames = []
for images, labels in dataloader:
    # 进行模型训练，并将文件名存储到列表中
    for i in range(len(images)):
        filename = dataset.get_filename(i)
        filenames.append(filename)