TF/Keras 'import_from_directory'导致目录NotFoundError

7y4bm7vi  于 2023-08-06  发布在  其他
关注(0)|答案(1)|浏览(93)

我正在Tensorflow/Keras中为CNN预处理数据。我使用image_dataset_from_directory方法从指定的目录中获取数据:

dataset = tf.keras.preprocessing.image_dataset_from_directory(data, seed=123,
    shuffle=True,
    image_size=(IMAGE_SIZE,IMAGE_SIZE),
    batch_size=BATCH_SIZE)

字符串
但它不是不从目录中获取数据,而是给出一个目录NotFoundError

NotFoundError           Traceback (most recent call last)
----> 1 dataset = tf.keras.preprocessing.image_dataset_from_directory(data, seed=123,
    shuffle=True,
    image_size=(IMAGE_SIZE,IMAGE_SIZE),
    batch_size=BATCH_SIZE)

截图

x1c 0d1x的数据
我试图通过手动指定数据的路径来纠正错误,但错误仍然存在。如何正确确定image_dataset_from_directory函数的数据路径?
或者是否有任何其他可能的建议,说明如何进行?

lpwwtiir

lpwwtiir1#

image_dataset_from_directory不是Keras preprocessing包的函数,而是utils包的函数。
您只需将调用替换为tf.keras.utils.image_dataset_from_directory即可。
文档:https://www.tensorflow.org/api_docs/python/tf/keras/utils/image_dataset_from_directory

相关问题