TypeError: Invalid shape of image data (3,32,32)

当你遇到这个错误时，说明你的图像数据的形状无效。

可能的原因

• 图像数据的形状不符合要求。一般情况下，图像数据的形状应该是 (height, width, channels)，即图片的高度、宽度和通道数。出现这个错误的原因可能是你的图像数据的形状并不符合这个规则。

• 图像数据的维度数不正确。在这个错误信息中，你可以看到 3，这是指图像数据只有三个维度，而实际上应该有四个维度。这可能是因为你需要将 batch size 添加到图像数据中。

解决办法

• 确认图像数据的形状是否正确。如果不正确，则需要重新调整图像数据的形状，使其符合规则。如下面的例子所示：

import numpy as np
from PIL import Image

# 加载图片
im = Image.open('image.png')

# 将图片转换成 numpy 数组
im_arr = np.array(im)

# 改变图片的形状
im_arr = np.transpose(im_arr, (2,0,1))

# 检查图片的形状
print(im_arr.shape)

在这个例子中，我们将图片的形状从 (height, width, channels) 转换成了 (channels, height, width)。

• 确认图像数据的维度数是否正确。如果维度数不正确，则需要将 batch size 添加到图像数据中。如下面的例子所示：

import numpy as np
from PIL import Image

# 加载图片
im = Image.open('image.png')

# 将图片转换成 numpy 数组
im_arr = np.array(im)

# 添加 batch size
im_arr = np.expand_dims(im_arr, axis=0)

# 检查图片的形状
print(im_arr.shape)

在这个例子中，我们将图像数据添加了一个维度，即把原来的 (height, width, channels) 转换成了 (1, height, width, channels)。

希望以上的解决办法能够帮助你解决这个问题。