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

关键代码

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 类型为其他类型

你可能感兴趣的:(python,笔记,pytorch,深度学习,人工智能)