调用:plt.imshow(img, cmap=“gray”) 时出错:

TypeError: Invalid shape (1, 28, 28) for image data

这是因为这个image是三维的,我们可以改为:

plt.imshow(img.squeeze(), cmap=“gray”)

.squeeze()函数可以把三维数组变为二维。因为它会把为1的维度给去掉

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐