python - 如何在Keras中获取由ImageDataGenerator的.flow_from_directory函数扫描的类的名称？

我想创建一个用户友好的GUI图像分类器，用户只需指向数据集的目录即可对模型进行训练，然后他们可以将任何图像提供给程序，它将显示概率和标签。图像中的对象。但是，如何获取Keras中ImageDataGenerator的.flow_from_directory函数扫描的类的名称？

最佳答案

从文档中，“可以通过属性class_indices获得包含从类名到类索引的映射的字典。”

https://keras.io/preprocessing/image/#flow_from_directory

在下面的示例中，train_data_dir包含两个子文件夹cat和dog。

train_datagen = ImageDataGenerator(rescale=1. / 255)

train_generator = train_datagen.flow_from_directory(
    train_data_dir,
    target_size=(img_width, img_height),
    color_mode='grayscale',
    shuffle = True,
    batch_size=batch_size,
    class_mode='binary')

print(train_generator.class_indices)
{'cat': 0, 'dog': 1}

`