原始 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')
