dataTtransform = transforms.Compose([transforms.ToTensor(),transforms.Resize([imageSize, imageSize]), ])

在定義transforms.Compose()時，把ToTensor()方法寫在了Resize()之前，造成順序不對。

修改如下：

dataTtransform = transforms.Compose([transforms.Resize([imageSize, imageSize]),transforms.ToTensor(), ])

總結

以上是生活随笔為你收集整理的img should be PIL Image. Got ＜class ‘torch.Tensor‘＞的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。