我将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。但显示相同的错误