python - 哪个 cmap (colormap) 与 matplotlib imshow() 一起使用，将 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')