如何在 ML Kit Translation 中获取下载的模型

Question

我正在尝试检查是否已下载 model 以在 ML Kit Translation 中进行翻译。 我没有找到这方面的任何官方文档，而是编写了查看来自 Google 的示例应用程序的代码。

此代码检查下载的模型中是否有正确的模型：

public static boolean isLanguageDownloaded(String language) {
    boolean isDownloaded = false;
    for (int i = 0; i < getAvailableModels().size(); i++) {
        if (getAvailableModels().get(i).equals(language)) {
            isDownloaded = true;
        } else {
            isDownloaded = false;
        }
    }
    return isDownloaded;
}

此代码可用于 model：

public static List<String> getAvailableModels() {
    List<String> availableModels = new ArrayList<>();
    getRemoteModelManager()
            .getDownloadedModels(TranslateRemoteModel.class)
            .addOnSuccessListener(
                    models -> {
                        // Model downloading is complete.
                        // ...
                        for (TranslateRemoteModel model : models) {
                            availableModels.add(model.getLanguage());
                        }
                    })
            .addOnFailureListener(
                    e -> {
                        // Model downloading failed.
                        // ...
                        Log.d(TAG, "onFailure: Model downloading failed.");
                    });
    return availableModels;
}

我真的很感激任何帮助和建议。

Answer 1

您可以为特定语言创建一个 TranslateRemoteModel，然后使用 RemoteModelManager#isModelDownloaded API 进行检查。 https://developers.google.com/android/reference/com/google/mlkit/common/model/RemoteModelManager#isModelDownloaded(com.google.mlkit.common.model.RemoteModel)