是否可以从tflite模型中提取训练好的类名？

Question

我尝试到处搜索，尝试了

tflite_interpreter = tf.lite.Interpreter(model_path='model.tflite')

中的所有内容，阅读了tflite文档，但我找不到从模型中提取类名的方法。

可以吗？

Answer 1

当您创建训练数据时，您可以将标签编码为数字表示。之后标签就会丢失，如果您知道训练模型所用的数据，那么您就可以知道 label_names。 label_names 的信息不存储在模型中。