是否可以手动创建 labels.txt?
Is it possible to create labels.txt manually?
我最近将我的模型转换为 tensorflow lite,但我只有 .tflite 文件,而不是我的 Android 项目的 labels.txt。那么是否可以使用我用来分类的 类 创建我自己的 labels.txt 呢?如果没有,那么如何生成labels.txt?
您应该能够生成并使用自己的 labels.txt。该文件需要按照您在训练中提供的顺序包含标签名称,每行一个名称。
运行安装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 或本地 运行 上,标签文件将使用每个子文件夹中的类别自动生成。
我最近将我的模型转换为 tensorflow lite,但我只有 .tflite 文件,而不是我的 Android 项目的 labels.txt。那么是否可以使用我用来分类的 类 创建我自己的 labels.txt 呢?如果没有,那么如何生成labels.txt?
您应该能够生成并使用自己的 labels.txt。该文件需要按照您在训练中提供的顺序包含标签名称,每行一个名称。
运行安装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 或本地 运行 上,标签文件将使用每个子文件夹中的类别自动生成。