小编der*_*Bai的帖子

通过tensorflow-datasets，我将 MNIST 数据集集成到 Tensorflow 中，现在想要使用 Matplotlib 可视化单个图像。我根据本指南做到了： https: //www.tensorflow.org/datasets/overview

不幸的是我在执行过程中收到一条错误消息。但它在指南中效果很好。

根据指南，您必须使用 take() 函数创建一个仅包含一张图像的新数据集。然后在指南中访问这些功能。在我的尝试过程中，我总是收到一条错误消息。

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import matplotlib.pyplot as plt
import numpy as np
import tensorflow.compat.v1 as tf

import tensorflow_datasets as tfds



mnist_train, info = tfds.load(name="mnist", split=tfds.Split.TRAIN, with_info=True)
assert isinstance(mnist_train, tf.data.Dataset)

mnist_example = mnist_train.take(50)

#The error is raised in the next line. 
image = mnist_example["image"]
label = mnist_example["label"]

plt.imshow(image.numpy()[:, :, 0].astype(np.float32), cmap=plt.get_cmap("gray"))
print("Label: %d" % label.numpy())

这是错误消息：

Traceback (most recent call …