面临错误 'DirectoryIterator' 对象没有属性 'cache'

Question

我正在编写代码来为 CNN 模型的性能配置数据集。不幸的是，我遇到了一个错误。

我的代码：

AUTOTUNE = tf.data.AUTOTUNE

train_data = train_data.cache().shuffle(1000).prefetch(buffer_size=AUTOTUNE)
valid_data = valid_data.cache().prefetch(buffer_size=AUTOTUNE)

错误：'DirectoryIterator' object has no attribute 'cache'

Answer 1

您可以使用

导入您的数据集

tf.data.Dataset.from_generator()

Answer 2

您可能会使用 flow_from_directory 和 ImageDataGenerator() 来设置数据集，但它们与 tf.data.AUTOTUNE 不兼容。 使用 tf.keras.preprocessing.image_dataset_from_directory 来加载数据集。 https://www.tensorflow.org/tutorials/images/classification