如何从 keras 模型制作 label.txt

Question

我在两个 类 上训练了 keras 模型。然后我通过以下代码将其转换为 .tflite：

# Converting a SavedModel to a TensorFlow Lite model. 
saved_model_dir = r'C:\Users\Munib\New folder\my_model.h5' 
loaded_model = tf.keras.models.load_model(saved_model_dir)
converter = tf.lite.TFLiteConverter.from_keras_model(loaded_model)# .from_saved_model(saved_model_dir) 
tflite_model = converter.convert()
open("my_model_converted_model.tflite", "wb").write(tflite_model)

当我在互联网上进一步研究如何在 android studio 上导入 .tflite 模型时。他们说我必须有两个文件：

1. my_model_converted_model.tflite

2. label.txt

我现在对如何从 keras 制作标签感到困惑，谁能告诉我谢谢

Answer 1

如果您的模型有 2 个 类，您模型的输出可能如下所示——例如：[0.12, 0.88]。如果第一个输出，即 0.12，对应于预测为“是”的概率，第二个输出，即 0.88，对应于预测为“现在”的概率，则创建名为“labels.txt”内容为：

Yes
No

就是这样！就这么简单:)

如果这解决了您的问题，请将其标记为已接受的答案:)