量化 TFLite 图像分类模型是否有通用的 label.txt 文件？

【问题标题】：Is there a common label.txt file for quantized TFLite image classification models?量化 TFLite 图像分类模型是否有通用的 label.txt 文件？
【发布时间】：2022-01-07 15:07:01
【问题描述】：

我使用来自 Tensorflow.org/lite 的入门模型构建了一个 android ML 图像分类应用程序。入门模型包括 .TFLite 模型以及随附的 label.txt 文件。我现在想在这个应用程序中尝试不同的 ML 模型。我已经下载了一些 inception 和 mobilenet 模型，但它们都没有附带 label.txt 文件。有谁知道是否有一个通用的 label.txt 文件可以与任何量化的 ML 模型一起使用，如果有，我可以在哪里获得副本？还是我唯一的 label.txt 文件适合所有人？或者，是否有人可以与我共享初始 ML 模型的 label.txt 文件？

【问题讨论】：

标签： tensorflow label tensorflow-lite image-classification tensorflow-hub

【解决方案1】：

文本文件的每一行都是模型的标签。因此，您可以使用标签更改旧 label.txt 文件的“标签”。您可以在入门代码中搜索如何使用此文件。使用支持库检查此link 以轻松加载 label.txt 文件。

使用 app build.gradle 文件中的依赖项。（查看版本here）

implementation 'org.tensorflow:tensorflow-lite-support:0.3.0'

在您的代码中：

import org.tensorflow.lite.support.common.FileUtil;

final String ASSOCIATED_AXIS_LABELS = "labels.txt";
List<String> associatedAxisLabels = null;

try {
   associatedAxisLabels = FileUtil.loadLabels(this, ASSOCIATED_AXIS_LABELS);
} catch (IOException e) {
   Log.e("tfliteSupport", "Error reading label file", e);
}

如果您需要其他任何事情，请联系我。

最好的

【讨论】：