小编cod*_*ord的帖子

我正在使用mnist数据集来在keras背景中训练胶囊网络。训练后，我想显示mnist数据集中的图像。为了加载图像，使用了mnist.load_data（）。数据存储为（x_train，y_train），（x_test，y_test）。现在，为了可视化图像，我的代码如下：

img_path = x_test[1]  
print(img_path.shape)
plt.imshow(img_path)
plt.show()

该代码给出的输出如下：

(28, 28, 1)

和在plt.imshow（img_path）上的错误如下：

TypeError: Invalid dimensions for image data

如何以png格式显示图像。救命！