报错: ‘img should be PIL Image. Got <class ‘torch.Tensor‘>’及解决方案

1. 问题描述

在定义transforms.Compose()时,把ToTensor()方法写在了resize的方法之前,造成顺序不对
报错: ‘img should be PIL Image. Got <class ‘torch.Tensor‘>’及解决方案_第1张图片

2. 解决方案:

将ToTensor()方法移动到resize之后

trans = transforms.Compose([transforms.ToTensor(),
                            transforms.Resize(224)])

在这里插入图片描述
换成

trans = transforms.Compose([transforms.Resize(224),
                            transforms.ToTensor()])

在这里插入图片描述

你可能感兴趣的:(Debug,python,pytorch)