axes don't match array in pytorch

最新推荐文章于 2023-10-12 10:27:01 发布

六十英里

最新推荐文章于 2023-10-12 10:27:01 发布

阅读量9.2k

点赞数 4

分类专栏：错误备忘录文章标签： torch torchvision Tensor

本文链接：https://blog.youkuaiyun.com/li_ellin/article/details/103288858

版权

在从numpy数组转换到PyTorch张量时遇到了`axes don't match array`错误。问题出在将变换应用到numpy数组列表而不是PIL图像上。解决方案是直接使用`torch.Tensor`进行类型转换，并在必要时使用`unsqueeze(0)`来调整维度。这种方法在python2.7和python3.7的PyTorch不同版本中都能得到相同效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一个有意思的错误：

self.transformer = transforms.Compose([transforms.ToTensor()])
img = self.transformer(img)
该语句在python3.7中运行无问题，在python2.7中出现如下错误：

img = torch.from_numpy(pic.transpose((2, 0, 1)))
ValueError: axes don't match array

解决方案：
错误原因：You are applying the transformation to a list of numpy arrays instead of to a single PIL image (which is usually what ToTensor() transforms expects).（来源：https://stackoverflow.com/questions/53995708/axes-dont-match-array-in-pytorch）
代码修改：
img = torch.Tensor(img/255.0)

img = img.unsqueeze(0)

用上述语句替换img = self.transformer(img)这个语句。
1. 因为transforms.ToTensor()完成了类型转换+归一化两个步骤，而torch.Tensor仅完成类型转换。
2. 我的代码中因为img是灰度图像，大小H*W，使用t