python - 有没有一种方法可以将不同类的图像全部放在一个目录中，以及它们对应的类到 Keras 中？

Question

我正在尝试对五个类的混合数据集执行图像分类。所有图像都在同一个目录中。我想使用 keras.preprocessing 的方便的image_dataset_from_directory方法。但由于文件都在一个目录中，它只检测所有文件的一个类。我还使用标签参数将其设置为长度列表，该列表的长度等于我包含标签的图像数量，但它只在一个类中读取它们。

train_ds = tf.keras.preprocessing.image_dataset_from_directory(
    "trainset",
    labels = trainlabels,
    image_size=image_size,
    batch_size=batch_size,
)



Found 4987 files belonging to 1 classes.

标签： pythontensorflowkeras