在Keras中如何获得现有模型的{class_indices`或预测标签

时间:2018-05-06 00:48:26

标签: keras prediction

我知道keras提供了一个.class_indicies字典,其中包含从.flow_from_directory()类(https://keras.io/preprocessing/image/ImageDataGenerator foreach$sql = "SELECT photoprofile,username from photo WHERE username IN ('somearray')"; $resol = array(); $resulol = mysqli_query($con,$sql); $photos = mysqli_fetch_all($resulol, MYSQLI_ASSOC); $photos = array_column($photos, "photoprofile", "username"); foreach ( $restest as $user ) { if ( isset($photos[$user])) { $res[] = $photos[$user]; } else { $res[] = ''; }; } 的一部分从类名到类的映射。

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

1 个答案:

答案 0 :(得分:0)

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

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

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

train_generator_example.class_indices。