Pytorch:TypeError: pic should be PIL Image or ndarray. Got <class ‘torch.Tensor‘>

许天逸
2023-12-01

关键代码

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()转换类型,因此会报错。

解决办法

删除 transforms.ToTensor()
或者
修改x 类型为其他类型

 类似资料:

相关阅读

相关文章

相关问答