Python 有没有一种方法可以将不同类的图像(都在一个目录中)及其对应的类提供给Keras?

Python 有没有一种方法可以将不同类的图像(都在一个目录中)及其对应的类提供给Keras?,python,tensorflow,keras,Python,Tensorflow,Keras,我正在尝试对五个类的混合数据集执行图像分类。所有图像都在同一目录中。我想使用keras.preprocessing的方便的image\u dataset\u from\u directory方法。但是由于文件都在一个目录中,所以它只为所有文件检测一个类。我还使用labels参数将其设置为长度等于包含标签的图像数量的列表,但它只在一个类中读取它们 train_ds = tf.keras.preprocessing.image_dataset_from_directory( "tr

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

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.


keras需要一些东西来定义哪个图像属于哪个类。你需要提供这些信息。是什么阻止您将它们放入不同的文件夹中?实际上,我使用labels参数提供了这些信息。但还是没有运气。我也使用了shutil和os将它们放在不同的目录中,但如果能找到解决方法,那就太好了。