如何使用keras在神经网络中打印预测类的名称？

Question

我在keras中使用预先训练的模型并最终预测类索引是一些整数值，但我似乎不明白如何打印这些类的名称？

我使用的模型是ResNet 50。

编辑：仍然无法弄明白，对我来说，让它回答编辑它真的很重要。

Answer 1

查看https://martin-thoma.com/image-classification/

关键部分是：

from imagenet_utils import preprocess_input, decode_predictions

preds = model.predict(x)
print('Predicted:', decode_predictions(preds))

Answer 2

如果您没有使用imagenet数据，而是在自己的设备上重新训练：使用flow_from_directory的{​​{1}}时，可以方便地将文件夹的标签保存在ImageDataGenerator可访问的位置在dict内！ dict的值对应于.class_indices输出的位置。

原油示例：

model.predict

另请参阅：https://keras.io/preprocessing/image/