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

Question

以前我收到一个错误

from tensorflow.examples.tutorials.mnist import input_data因为它不再是 TensorFlow 2.0 的一部分

这篇文章中建议我将其替换为

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 的库已更改，其功能也随之更改。因此，read_data_sets不是必需的，并且不起作用。input_data已经是包含image和label字典的数据集。您可以像访问一样

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

阅读此处了解更多信息。