tensorflow - Keras image_dataset_from_directory 未找到图像

Question

我想从我有大约 5000 张图像的目录中加载数据(类型“png”)。但是它返回一个错误，说当明显有图像时没有图像。
这段代码:

width=int(wb-wa)
height=int(hb-ha)
directory = '/content/drive/My Drive/Colab Notebooks/Hair/Images'
train_ds = tf.keras.preprocessing.image_dataset_from_directory(
    directory, labels=densitat, label_mode='int',
    color_mode='rgb', batch_size=32, image_size=(width, height), shuffle=True, seed=1,
    validation_split=0.2, subset='training', follow_links = False)

返回:

ValueError: Expected the lengths of `labels` to match the number of files in the target directory. len(labels) is 5588 while we found 0 files in /content/drive/My Drive/Colab Notebooks/Hair/Images.

我可以看到图像:
Colab view of the folder structure with the images
问题出在哪儿？我需要使用这个函数来批量加载数据，因为我有一个很大的数据集

Answer 1

我找到了答案，所以我发帖以防它可能对某人有所帮助。
问题是路径，因为我使用的是带有图像的文件夹的路径，而我应该使用目录(上面的一个文件夹)。

directory = '/content/drive/My Drive/Colab Notebooks/Hair'

请注意，'/Hair' 是包含我的图像的文件夹。