哪个 cmap(颜色映射)与 matplotlib imshow() 一起使用以将 MNIST 数据集图像显示为白色背景上的黑色字符

Which cmap (colormap) to use with matplotlib imshow() to diplay the MNIST dataset images as black characters on white background

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

只是为了看东西,我使用 matplotlib 来显示数据集中的示例,但是当我选择 'gray' 的 cmap 时,我得到黑色背景和白色字符,如下所示。

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

是否有其他颜色图可以在白色背景上正确显示黑色图像?

您可以 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')