Как использовать customvision.ai для создания модели обнаружения объектов для TensorFlow Lite? - PullRequest
0 голосов
/ 09 октября 2019

У меня есть модель обнаружения объекта, которую я создал в https://customvision.ai. Если я экспортирую ее как модель TensorFlow Lite, я получу модель, которая ожидает FLOAT32 [1, 416, 416, 3] в качестве ввода и возвращает FLOAT32 [1, 13, 13, 35] в качестве вывода (в соответствии с TensorFlow Lite visualize.py).

Я хотел бы использовать эту модель в приложении для Android. Я пытался загрузить файл модели .tflite в пример приложения *1000* TensorFlow Lite , однако он ожидает другой формат. Я получаю следующее исключение при запуске приложения. java.lang.IllegalArgumentException: Cannot copy between a TensorFlowLite tensor with shape [1, 13, 13, 35] and a Java object with shape [1, 10, 4].

Возможна ли адаптация примера приложения для использования модели из customvision.ai?

Как следует интерпретировать форму [1, 13, 13, 35]?

Спасибо зазаранее!

...