具体报错:
TypeError: batch must contain tensors, numbers, dicts or lists; found <class ‘PIL.Image.Image’>
1
2
loader的代码:
dataloader=torch.utils.data.DataLoader(dataset,batch_size=1,shuffle=True)
1
表面上看这个代码没有问题,实际上问题出在了dataloader机制的要求上,dataloader要求接收的是一个tensor,而我的dataset没有做transform,所以dataset的getitem函数返回的是一个PIL的Image对象,所以就会报错
作者:GZKPeng
来源:CSDN
原文:https://blog.csdn.net/zkp_987/article/details/81078923
版权声明:本文为博主原创文章,转载请附上博文链接!