如何从 keras 模型制作 label.txt答案

【问题标题】：How to make label.txt from keras model如何从 keras 模型制作 label.txt
【发布时间】：2020-10-11 05:50:26
【问题描述】：

我在 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 模型时。他们说我必须有两个文件：

my_model_converted_model.tflite
标签.txt

我现在对如何从 keras 制作标签感到困惑，有人可以告诉我吗？谢谢

【问题讨论】：

各位，这是我关注的网站：codelabs.developers.google.com/codelabs/…
是的，我正在学习相同的教程，我在 android 应用程序开发方面有经验，但我对 Tensorflow 和 ML 完全陌生，我不知道从哪里下载这两个文件？我确实运行时运行了所有.. 还从 android studio 中的项目开始添加了依赖项，但我真的不知道从哪里下载这个模型和其他文件？请帮助..

标签： python android-studio tensorflow keras tf.keras

【解决方案1】：

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

Yes
No

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

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

【讨论】：