keras - 在 Keras 中如何获取现有模型的 `class_indices` 或预测标签

Question

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

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

Answer 1

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

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

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

train_generator_example.class_indices。