Использование моделей API обнаружения объектов TensorFlow при прогнозировании

Я использовал модель API обнаружения объектов SSD TensorFlow. с нуля. Сценарий оценки показывает, что модель чему-то научилась, и теперь я хочу использовать модель.

Я изучил функцию обнаружения объектов ipynb, которая может подавать отдельные изображения в обученную модель. Однако это для SSD с MobileNet. Я использовал следующую строку (после загрузки мета-графика) для вывода тензорных имен модели TensorFlow, которую я обучал.

print([str(op.name) for op in tf.get_default_graph().get_operations()] )

Но он не содержит тех же имен тензоров ввода или вывода, что и в ipynb. Я также просмотрел код, но многие функции указывают друг на друга, и мне трудно найти то, что я ищу.

Как мне найти нужные мне тензорные имена? Или есть еще один метод, о котором я не знаю?


person Cassie    schedule 18.10.2017    source источник


Ответы (1)


Чтобы использовать график, вам необходимо заморозить / экспортировать его с помощью предоставленный скрипт. Полученный файл .pb будет содержать нужные вам узлы. Не знаю, почему это так организовано, но это так.

person gdelab    schedule 18.10.2017