python - Pytorch：如何使用参考表创建自定义数据集

Question

我有一个 reference.csv 文件，它包含三列：类型、类和路径。以下是前 5 个示例行：

"Type","Class","Path"
"train","A","./path1/001.jpg"
"train","A","./path2/002.jpg"
"test","C","./path3/003.jpg"
"train","B","./path4/001.jpg"
"test","B","./path5/002.jpg"
...

以更易于查看的格式：

|----------------------|------------------|------------------|
|         Type         |       Class      |       Path       |
|----------------------|------------------|------------------|
|        train         |         A        | ./path1/001.jpg  |
|----------------------|------------------|------------------|
|        train         |         A        | ./path2/002.jpg  |
|----------------------|------------------|------------------|
|        train         |         C        | ./path3/003.jpg  |
|----------------------|------------------|------------------|
|        test          |         B        | ./path4/001.jpg  |
|----------------------|------------------|------------------|
|        test          |         B        | ./path5/002.jpg  |
|----------------------|------------------|------------------|

我想创建一个数据集类 (torch.utils.data.Dataset) 来读取图像，这样我就可以使用 DataLoader (torch.utils.data.DataLoader)。

使用参考表创建自定义数据集的正确方法是什么？