Обнаружение объекта TF Lite только возвращает 10 обнаружений - PullRequest
0 голосов
/ 22 сентября 2019

Я использую пользовательскую модель обнаружения объектов с TensorFlow Lite на родном Android.На данный момент я обнаружил только 2 пользовательских объекта.Я использую TensorFlow Object Detection API, и у меня есть конвейер, который создает оптимизированные файлы .tflite.

Однако во время вывода модель возвращает только до 10 отдельных обнаружений.Согласно https://www.tensorflow.org/lite/models/object_detection/overview, это ожидается.Проблема в том, что мои изображения имеют относительно большую плотность объектов.Мне нужно иметь возможность обнаруживать до 30 отдельных объектов на один вызов изображения / вывода.

Если я изменю NUM_DETECTIONS в образце приложения Android из репозитория TF с 10 на, скажем, 20, я получу исключение времени выполненияиз-за несоответствия формы.Как я могу создать файлы .tflite, способные дать более 10 экземпляров обнаружения объектов?

Спасибо!

...