Python 如何将图像和标签都提供给pytorch中的DataLoader?

Python 如何将图像和标签都提供给pytorch中的DataLoader?,python,pytorch,Python,Pytorch,我有一个二维张量,它包含我的帧信息,还有一个一维张量,它包含与每个帧相关的输出或标签信息。我创建一个函数并返回(帧、标签)。然后,我将这两个参数赋给“torch.utils.data.DataLoader()”。但是数据加载器不能同时对这两种(帧、标签)工作,只能对其中一种进行迭代。这很奇怪。你能展示一下你的数据集类的getitem\uuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuu方法如何返回帧和标签吗?我使用了灰色图像,制作了每个帧的1D数组,并附加了它们。因

我有一个二维张量,它包含我的帧信息,还有一个一维张量,它包含与每个帧相关的输出或标签信息。我创建一个函数并返回(帧、标签)。然后,我将这两个参数赋给“torch.utils.data.DataLoader()”。但是数据加载器不能同时对这两种(帧、标签)工作,只能对其中一种进行迭代。

这很奇怪。你能展示一下你的
数据集
类的
getitem\uuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuu方法如何返回
帧和
标签吗?我使用了灰色图像,制作了每个帧的1D数组,并附加了它们。因此,我有一组2D输入,其中包含我的图像。函数返回(帧、标签)。我将函数输出给Dataloader.def read_dataset():num_train=2000 yuv=np.load('H200_train_dataset.npy')label=np.load('Cmpr.npy')train_img,train_labels=yuv[:num_train,:],label[:num_train,:]test_labels=yuv[num_train:num_train:num train+200,:],label[num_train:num_train:num-train+200,:]返回(列车img,列车标签),(测试img,测试标签)请不要在逗号中发布代码-它无法读取。请编辑您的问题并在那里添加正确格式的代码。这很奇怪。请您展示一下您的
数据集
类的
getitem\uuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuuu>方法如何返回
标签
?我使用了灰色图像,并制作了每一个1D数组并附加它们。so、 我有一组包含图像的2D输入。函数返回(帧、标签)。我将函数输出给Dataloader.def read_dataset():num_train=2000 yuv=np.load('H200_train_dataset.npy')label=np.load('Cmpr.npy')train_img,train_labels=yuv[:num_train,:],label[:num_train,:]test\u img,test\u labels=yuv[num\u train:num\u train+200,:],label[num\u train:num\u train+200,:]return(train\u img,train\u labels),(test\u img,test\u labels)请不要用逗号发布代码-它无法读取。请编辑您的问题并在那里添加代码,格式正确。