我想创建一个用户友好的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}
`
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句