面临错误“DirectoryIterator”对象没有属性“缓存”答案

【问题标题】：Facing error 'DirectoryIterator' object has no attribute 'cache'面临错误“DirectoryIterator”对象没有属性“缓存”
【发布时间】：2021-04-25 12:08:23
【问题描述】：

我正在编写代码来配置数据集以提高 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'

【问题讨论】：

哪一行报错了？
在下面一行train_data = train_data.cache().shuffle(1000).prefetch(buffer_size=AUTOTUNE)@NicholasHunter

标签： python tensorflow keras conv-neural-network

【解决方案1】：

您可以使用

导入数据集

tf.data.Dataset.from_generator()

【讨论】：

【解决方案2】：

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

【讨论】：