Teachable Machine生成的TensorFlow Lite模型DisplayName为空求助

问题确认 #

你提到用Teachable Machine生成的TFLite模型部署到Android后，检测结果的displayName为空，但index、score等参数正常，且训练时已设置"Cats"和"Dogs"作为分类名称，输出示例如下：

<Category "1" (displayName= score=0.99609375 index=1)>

以下是具体排查和解决步骤：

1. 检查模型元数据是否包含分类名称 #

Teachable Machine导出的TFLite模型，分类名称会存储在模型的元数据中。你可以通过以下方式验证：

• 使用本地的TensorFlow Lite Model Viewer工具打开下载的tflite文件，查看元数据内的标签列表是否存在"Cats"和"Dogs"。如果元数据中没有这些标签，说明导出过程中分类名称未被正确写入模型。

2. 重新导出模型并验证文件完整性 #

重新从Teachable Machine导出模型时，注意：

• 选择TensorFlow Lite格式，确认导出的压缩包中包含tflite模型文件和labels.txt标签文件（部分版本会自动打包两者）。
• 导出过程中确保网络稳定，避免因中断导致元数据缺失。导出后可以将模型重新导入Teachable Machine，确认分类名称能正常显示。

3. Android端手动映射标签的补救方案 #

如果模型确实未携带displayName元数据，可以通过手动加载标签文件来补全：

1. 在Android项目的assets目录下创建labels.txt，内容为：

Cats
Dogs

2. 在ImageClassifierHelper中添加标签加载逻辑：

private val labels: List<String> by lazy {
    context.assets.open("labels.txt").bufferedReader().readLines()
}

3. 在处理分类结果时，通过index匹配对应的标签：

// 在你的ClassifierListener实现中处理结果
override fun onResults(results: List<Classifications>?, inferenceTime: Long) {
    results?.forEach { classifications ->
        classifications.categories.forEach { category ->
            val displayName = if (category.index < labels.size) {
                labels[category.index]
            } else {
                "Unknown"
            }
            // 使用displayName进行后续逻辑处理
        }
    }
}

4. 排查训练阶段的分类名称设置 #

确认在Teachable Machine训练时，分类名称的输入没有特殊字符或空格（虽然一般不影响，但极端情况可能导致元数据写入异常），训练完成后再次检查分类列表是否正确显示"Cats"和"Dogs"。

内容的提问来源于stack exchange，提问作者BraveEvidence