首页 > 解决方案 > .lite 和 .tflite 格式有什么区别

问题描述

.lite.tflite格式(TensorFlow 格式)有什么区别?如果没有区别,为什么会有两个?

除了问题之外,我似乎无法将带有.lite扩展文件的模型上传到 Firebase ML 套件。这可能是什么原因?

标签: androidfirebasetensorflowtensorflow-litefirebase-mlkit

解决方案


ML 开发人员首先训练一个 TensorFlow 模型,然后使用TOCO将其转换为 TensorFlow Lite 模型。运行 TOCO 命令时,您可以为转换后的 Lite 模型指定任何输出名称。所有 TensorFlow Lite TOCO 示例都使用 .tflite 扩展名;但 .lite 似乎是人们愿意选择的另一个流行扩展名。

因此,只要它是 TensorFlow Lite FlatBuffer 格式的模型,TensorFlow Lite 就能够加载/运行模型,而不管扩展如何。

但不幸的是,此时 ML Kit Console 只接收扩展名为 .tflite 的文件。我们可以考虑取消该强制执行。同时,如果您确定它是 TensorFlow Lite 模型,只需重命名扩展并上传即可。


推荐阅读