调用:plt.imshow(img, cmap=“gray”) 时出错:
TypeError: Invalid shape (1, 28, 28) for image data
这是因为这个image是三维的,我们可以改为:
plt.imshow(img.squeeze(), cmap=“gray”)
.squeeze()函数可以把三维数组变为二维。因为它会把为1的维度给去掉。
调用:plt.imshow(img, cmap=“gray”) 时出错:
TypeError: Invalid shape (1, 28, 28) for image data
这是因为这个image是三维的,我们可以改为:
plt.imshow(img.squeeze(), cmap=“gray”)
.squeeze()函数可以把三维数组变为二维。因为它会把为1的维度给去掉。