无法遍历DataLoader对象

时间:2020-07-20 09:35:27

标签: python dataset pytorch torchvision dataloader

我将ucf 101数据集加载如下:

dataComplete = torchvision.datasets.UCF101('directory', frames_per_clip=16, step_between_clips=1, frame_rate=None, fold=1, train=True, transform=transformF, _precomputed_metadata=None, num_workers=1, _video_width=0, _video_height=0, _video_min_dimension=0, _audio_samples=0)
data = torch.utils.data.DataLoader(dataComplete, batch_size=3, shuffle=True)
#where transformF = transforms.ToTensor()

并且我试图访问存储在dataLoader对象中的张量,但是每次我尝试执行时:

for n_batch in data:
  #training...

我得到了错误

TypeError: pic should be PIL Image or ndarray. Got <class 'torch.Tensor'>

我尝试改用dataComplete并使用enumerate(),range()ecc。但显示相同的错误

0 个答案:

没有答案
相关问题