DataLoader.dataset.samples[i]返回的其他参数是什么

Question

我使用torch.utils.data.DataLoader在可迭代样式的数据集中查找文件名。当在迭代循环中调用DataLoader.dataset.samples[i]时，我会得到文件名和一个整数。示例：

('car.JPEG', 0)

第二个参数代表什么？

Answer 1

DataLoader返回dataset对象的每个成员，该对象是一个包含样本及其所属class_index的元组。因此，在您的情况下，元组的第一个索引包含样本(car.JEPG(，第二个索引是样本所属class_index(在您的例子中为0(。要了解更多信息，请查看文档。