将TFDS数据集转换为tensorflow.examples.tutorials.mnist

Question

之前，我收到一个错误

from tensorflow.examples.tutorials.mnist import input_data，因为它不再是TensorFlow2.0的一部分

我在this帖子中建议将其替换为

import tensorflow_datasets as tfds
input_data = tfds.load(name="mnist", split=tfds.Split.TRAIN)

但现在我得到了错误：

Traceback (most recent call last):
  File "GAN.py", line 79, in <module>
    mnist = input_data.read_data_sets('../../MNIST_data', one_hot=True)
AttributeError: '_OptionsDataset' object has no attribute 'read_data_sets'

如何将input_data转换为正确的格式，以便能够从中使用read_data_sets？

Answer 1

随着TF2.0的库发生了变化，它的函数也发生了变化。input_data已经是你的数据集，由image和label的字典组成。你可以像这样访问

image, label = input_data["image"], input_data["label"]

请阅读here了解更多信息。