Pytorch的Dataloader报错：TypeError: batch must contain tensors, numbers, dicts or lists

具体报错：

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
来源：优快云
原文：https://blog.youkuaiyun.com/zkp_987/article/details/81078923
版权声明：本文为博主原创文章，转载请附上博文链接！