Breaking News: Grepper is joining You.com. Read the official announcement!

TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'PIL.Image.Image'>

Bored Coder answered on April 19, 2020 Popularity 9/10 Helpfulness 10/10

answer TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'PIL.Image.Image'>

TypeError: default_collate: batch must contain tensors, numpy arrays, numbers, dicts or lists; found <class 'PIL.Image.Image'>

Comment

The error states that the DataLoader receives a PIL image. 
This is because there are no transforms made (transform=None) on the image.
You can add a transform that creates a tensor from the PIL image by adding transform:

from torchvision import transforms

transform = transforms.Compose([
    # you can add other transformations in this list
    transforms.ToTensor()
])

Popularity 9/10 Helpfulness 10/10 Language typescript

Source: discuss.pytorch.org

Tags: contain numpy typescript

Link to this answer
Share Copy Link