TensorFlow Lite奇怪的label.txt行为

时间:2019-11-28 18:18:32

标签: java python tensorflow tensorflow-lite

我正在尝试将此示例与自己的数据集结合使用:https://github.com/tensorflow/examples/blob/master/tensorflow_examples/lite/model_customization/demo/text_classification.ipynb

模型经过训练后,将其添加到Android项目中,如下所示:https://github.com/tensorflow/examples/tree/master/lite/examples/text_classification/android

model.export('text_classifier.tflite', 'text_label.txt', 'vocab.txt') 这行代码创建了包含以下子文件夹名称的text_label文件:

labels

我按照Google的示例将这些文件以及文件夹复制到了Android项目中。问题是它在我的Android项目中返回了错误的结果。即使完全错误,text_label.txt文件中最先出现的内容也更有可能。因此,如果第一个标签为label1,分类将始终说label1的概率为51%,label2为49%或类似的水平。如果label2是text_label.txt中的第一个标签,则相反。

如果我添加两个以上的标签,问题甚至更奇怪。

可能是什么问题?我如何解决它? Android示例本身有问题吗?

0 个答案:

没有答案