我想制作一个用户友好的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}