В Android Studio, после импорта модели tensorflow Lite, как можно использовать сгенерированный пример кода?

#android #android-studio #tensorflow #tensorflow-lite

#Android #android-studio #tensorflow #tensorflow-lite

Вопрос:

Через android Studio я добавил новый файл .tflite (в частности, тот, который можно найти здесь). Он сгенерировал следующий пример кода :

 try {
    LiteModelDeeplabv31Metadata2 model = LiteModelDeeplabv31Metadata2.newInstance(context);

    // Creates inputs for reference.
    TensorImage image = TensorImage.fromBitmap(bitmap);

    // Runs model inference and gets result.
    LiteModelDeeplabv31Metadata2.Outputs outputs = model.process(image);
    List<Category> segmentationMasks = outputs.getSegmentationMasksAsCategoryList();

    // Releases model resources if no longer used.
    model.close();
} catch (IOException e) {
    // TODO Handle the exception
}
 

LiteModelDeeplabv31Metadata2 невозможно найти. Это класс, который мне нужно создать, или его можно откуда-то импортировать?

Комментарии:

1. Удалось ли вам преодолеть проблему? Вам все еще нужна помощь?

2. казалось, все было в порядке после выхода и повторного открытия Android studio…im не знаете, как обновить в противном случае

Ответ №1:

LiteModelDeeplabv31Metadata2 должно быть автоматически сгенерировано, и у вас не должно быть никаких проблем с ссылками.

Убедитесь, что в вашем build.gradle есть следующий раздел

 buildFeatures {
  mlModelBinding true
}
 

Не могли бы вы проверить шаги в https://developer.android.com/studio/releases/#4.1-tensor-flow-lite-models ?

Также вот видео на YouTube о том, как его использовать https://www.youtube.com/watch?v=s_XOVkjXQbU