python-3.x - 从时尚 MNIST 数据打印图像给出彩色图像？

Question

著名的用于深度学习/计算机视觉的 fashion_MNIST 数据集是灰度图像[ https://github.com/zalandoresearch/fashion-mnist] .

该数据集已可用于 keras [ https://keras.io/datasets] .

但是，当我在 Tensorflow Keras API 中加载数据集并尝试打印一些图像时。我得到了彩色图像。我想知道为什么？谁能解释一下。这是我的代码:

import tensorflow as tf
mnist = tf.keras.datasets.fashion_mnist
(training_images, training_labels), (test_images, test_labels) = mnist.load_data()

import matplotlib.pyplot as plt
plt.imshow(training_images[0])
#print(training_labels[0])
#print(training_images[0])

Answer 1

它的发生是因为 matplotlib 的 imshow() 方法。默认情况下，它使用 rcParams["image.cmap"] 到 'viridis' 因此我们得到了彩色图像。你可以在这里看到 https://matplotlib.org/3.1.0/api/_as_gen/matplotlib.pyplot.imshow.html .您还可以通过运行以下代码在终端中检查这些参数:

 import matplotlib.pyplot as plt
 print(plt.rcParams)

如果我们将其设置为“Greys”，那么我们将得到灰度图像。您可以通过运行此代码来设置它:

plt.rcParams['image.cmap'] = 'Greys'

您可以在此处阅读有关“viridis”的更多信息:https://matplotlib.org/3.1.1/tutorials/colors/colormaps.html