android - 如何使用 TensorFlow Lite 在设备上使用多个自定义模型？

Question

我想用该应用程序运送多个自定义模型。所有模型都有相同的用途。我现在不想主持它。

自定义模型代码 github

FirebaseLocalModelSource localModelSource =
        new FirebaseLocalModelSource.Builder(LOCAL_MODEL_NAME)
                .setAssetFilePath(LOCAL_MODEL_PATH).build();
// add multiple

FirebaseModelManager manager = FirebaseModelManager.getInstance();
manager.registerLocalModelSource(localModelSource);

// access multiple

1. 我怎样才能访问多个模型？
2. 即使我托管它，那么我如何访问不同的自定义模型以实现相同的目的？

如果使用 mlkit 无法实现上述功能，是否有其他方法可以组合所有模型的结果？

Answer 1

我认为我们还不支持在一个推理中运行多个模型。如果你想在不同的场景下运行不同的模型，你可以给它们分配不同的名字，并使用不同的模型源来触发它们。

如果您可以更清楚地描述您的用例，我们可以看到将来如何支持它。

Answer 2

上面代码片段中的 LOCAL_MODEL_NAME 变量指向您与应用捆绑的模型文件。您只需将该值更改为指向与您的应用程序捆绑在一起的模型之一。

托管模型时，它的工作方式类似。每个模型都有不同的名称，因此您只需传递要使用的模型的名称。