与matplotlib imshow（）一起使用哪个cmap（colormap）将MNIST数据集图像显示为白色背景上的黑色字符

Question

原始MNIST图像是白色背景，上面有灰度字符。 0表示白色，255表示黑色，中间是灰色阴影。我正在使用keras提供的数据集副本，该副本使用相同的格式。

只是为了查看内容，我正在使用matplotlib来显示数据集中的示例，但是当我选择'gray'的cmap时，我会得到带有白色字符的黑色背景，如下所示。

  import matplotlib.pyplot as plt
  plt.imshow(X_train[0], cmap='gray')

是否有另一个色彩图可以在白色背景上正确显示图像？

Answer 1

您可以invert the colormap（使用色卡的*_r版本）或反转（否定）您的数据。

# Invert the colormap
plt.imshow(X_train[0], cmap='gray_r')

# Invert your data
plt.imshow(-X_train[0], cmap='gray')