Pytorch 自定义数据加载器的getitem不工作

我是新的深入学习和学习。我有6000个图像的数据集，在一个文件夹中有所有四个类。我有一个.csv文件，它有一个图像名和对应的一个热编码表示（）。我想把图像名称绑定到它的标签上，这样我就可以把它输入到神经网络中。我发现定制的数据加载器可以用于此。我在PyCharm中尝试了以下代码片段：

class-DFUDataset（数据集）：
def uuu init uuu（self，csv_文件，根目录，transform=None）：
self.DFU=pd.read\u csv（csv\u文件）
self.root\u dir=root\u dir
self.transform=transform
定义（自我）：
返回len（self.DFU）
def uu getitem uu（self，idx）：
如果火炬是张量（idx）：
idx=idx.tolist（）
img_name=os.path.join（self.root_dir，self.DFU.iloc[idx，0]）#映像名
image=io.imread（img\u名称）
img_label=os.path.join（self.root_dir，self.DFU.iloc[0，idx[1:]）#图像标签
示例={'image'：图像，'img_标签]：img_标签}
回样
DFU_数据集=DFUDataset(
csv\u file='C:/Users/aleems2/Desktop/dfu/DFUC2021\u列车组\u 210427/DFUC2021\u列车/标签\u数据\u地面\u真相.csv'，
root\u dir=“C:/Users/aleems2/Desktop/dfu/DFUC2021\u trainset\u 210427/DFUC2021\u train/labeled\u test\u images”）

如果idx不是张量，您希望它做什么？我想将.csv文件中的图像名称与该图像的标签相结合。我附加了原始问题中的.csv文件截图。@user1572349否问题是当

idx

不是张量时，您想做什么？因为这似乎是这里的问题。顺便说一句。你不需要考虑太多关于蓝点的事情。您可以通过添加

x=DFU\u数据集[0]