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

Question

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

只是看一下事情，我matplotlib用来显示数据集中的示例，但是当我选择的cmap时'gray'，会得到黑色背景和白色字符，如下所示。

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

是否有另一个颜色图可以将图像正确显示为白色背景上的黑色？

Answer 1

您可以反转颜色图（使用颜色图的*_r版本）或反转（否定）数据。

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

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