Как сделать так, чтобы обнаружение объектов тензорного потока работало быстрее на Android? - PullRequest
3 голосов
/ 15 марта 2019

У меня вопрос по поводу API обнаружения объектов Tensorflows . Я обучил Faster R-CNN Inception v2 модель своим собственным набором данных для классификации дорожных знаков и хочу развернуть его на Android, но API обнаружения объектов Tensorflows для Android и / или Tensorflow Lite поддерживает только модели SSD.

Есть ли способ развернуть модель Faster R-CNN на Android? Я имею в виду, как я могу поместить свой замороженный граф вывода Faster R-CNN в Android API вместо графического вывода SSD?

1 Ответ

3 голосов
/ 15 марта 2019

Для моделей с твердотельным накопителем это должно быть возможно с помощью инструмента export_tflite_ssd_graph.py.Но:

В настоящее время поддерживаются только модели SSD.Модели, такие как rapid_rcnn, в настоящее время не поддерживаются

Подробнее см. в этом руководстве .

...