是否可以手动创建labels.txt？

Question

我最近将我的模型转换为 tensorflow lite，但我只得到了 .tflite 文件，而不是我的 Android 项目的 labels.txt。那么是否可以使用我用来分类的类来创建我自己的 labels.txt？如果没有，那么如何生成labels.txt？

Answer 1

您应该能够生成和使用自己的 labels.txt。该文件需要按照您在训练中提供的顺序包含标签名称，每行一个名称。

Answer 2

安装TFLite Model Maker库后运行以下代码，传递数据集进行分类：

data = ImageClassifierDataLoader.from_folder('folder/')
train_data, test_data = data.split(0.8)
model = image_classifier.create(train_data)
loss, accuracy = model.evaluate(test_data)
model.export('image_classifier.tflite', 'imageLabels.txt')

在 Colab 或本地运行时，标签文件将使用每个子文件夹中的类别自动生成。