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

Question

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

Answer 1

根据文档，“包含从类名称到类索引的映射的字典可以通过属性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}

`