I load the dataset with the following transformations:
normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])

transform = transforms.Compose([
        transforms.Resize((240, 240), interpolation=0),
        transforms.ToTensor(),
        normalize
    ]),

Then when I try to convert the resulting tensor back to a PIL Image I get the following:
trans = transforms.ToPILImage(mode='RGB')
plt.imshow(trans(img.squeeze()))
plt.show()


Screenshot from 2018-09-28 10-25-14.jpg552×515 185 KB

Clearly, the image is not as it should be. Can anyone let me know what’s going on?
Thanks.