我正在编写代码来配置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'
2条答案
按热度按时间6vl6ewon1#
您可以导入数据集，
cld4siwp2#
您可能使用
flow_from_directory沿着和ImageDataGenerator()来设置数据集，但它们与tf.data.AUTOTUNE不兼容。使用tf.keras.preprocessing.image_dataset_from_directory加载数据集。https://www.tensorflow.org/tutorials/images/classification