我正在编写代码来配置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