Обнаружение Tensorflow Lite с помощью пользовательской модели AutoML

Question

Мне нравится тестировать пример обнаружения объектов TFLite.

https://github.com/tensorflow/examples/tree/master/lite/examples/object_detection/android

Пример с моделью по умолчанию отлично работает. Но я хочу проверить пользовательскую модель, сгенерированную из AutoML. Когда я заменяю «detect.tflite» и «labelmap fille» в каталоге «\ src \ main \ assets» и собираю, приложение запускается после запуска.

Моя модель очень проста ... может обнаружить только 2 объекта (тигр и лев). И мой файл labelmap содержит ниже:

??? Tiger Lion

Также я комментирую строку "// apply from: 'download_model.gradle'" в "build.gradle", чтобы остановить загрузку модели по умолчанию и использовать мою пользовательскую модель из ресурса.

Я новичок как в Android, так и в этом пространстве ML. Я буду рад, если кто-нибудь сможет посоветовать App cra sh после запуска с пользовательской моделью AutoML.

Заранее спасибо.

С уважением.

Answer 1

Две возможные ошибки в журнале могут быть:

Cannot convert between a TensorFlowLite buffer with 1080000 bytes and a ByteBuffer with 270000 bytes. Изменить TF_OD_API_INPUT_SIZE соответственно.

tflite ml google [1, 20, 4] and a Java object with shape [1, 10, 4]. Изменить NUM_DETECTIONS в соответствии с пользовательской моделью.