TypeError: img should be PIL Image. Got <class ‘torch.Tensor‘>

出现报错如图:

问题原因:

在定义transforms.Compose()时,把ToTensor()方法写在了随机旋转图片的方法RandomHorizontalFlip()之前,造成顺序不对

解决方法:将ToTensor()方法移动到RandomHorizontalFlip()之后

TypeError: img should be PIL Image. Got <class ‘torch.Tensor‘>_第1张图片

此时运行程序,报错消失。 

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