关键代码
x = torch.randn(64,3,128,128)
transform=transforms.Compose(
[transforms.Resize(64), transforms.ToTensor(), transforms.Normalize([0.5], [0.5])]
)
transform(x)
原因
在于 x 本就是 Tensor 类型的,有写了一次ToTensor()转换类型,因此会报错。文章来源:https://www.toymoban.com/news/detail-555206.html
解决办法
删除 transforms.ToTensor()
或者
修改x 类型为其他类型文章来源地址https://www.toymoban.com/news/detail-555206.html
到了这里,关于Pytorch:TypeError: pic should be PIL Image or ndarray. Got <class ‘torch.Tensor‘>的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!