在 Keras 中，如何获取现有模型的“class_indices”或预测标签

Question

我知道 keras 提供了一个 .class_indicies 字典，其中包含从类名到类的映射，作为其 ImageDataGenerator 类 (https://keras.io/preprocessing/image/) 中 .flow_from_directory() 的一部分。

但是，有没有办法从现有的 保存 模型（保存为 .h5 文件的模型）访问相应的类标签？这在将我的模型投入生产和提供预测服务时似乎很重要，因为预先不知道这些类，因此图像不会在预先标记的目录中分开。

Answer 1

它在火车生成器的 class_indices 属性中。

train_generator_example = ImageDataGenerator.flow_from_dataframe( 数据框， 目录=无， ………… )

标签可以从作为“train_generator_example”属性的字典中获取

train_generator_example.class_indices。