Как узнать правильную операцию ввода и вывода замороженного графа тензорного потока?

Получил нейронную сеть, обученную на python, формат - замороженный граф (.pb). Чтобы сделать вывод с использованием C-API (который требуется в этом особом случае), мне нужно узнать имя операции ввода и вывода графа. Очевидно, имена отличаются от тех, что были во время обучения. График содержит 1780 операций, в названии которых содержится слово «вход» или «выход»:

Вход:

bn1a_branch2a/keras_learning_phase/input

input_1_7



output:

output_1/kernel

output_1/bias

output_1_1/kernel

output_1_1/bias

output_1_2/kernel

output_1_2/bias

output_1_3/kernel

output_1_3/bias

output_1_4/kernel

output_1_4/bias

output_1_5/kernel

output_1_5/bias

output_1_6/kernel

output_1_6/bias

output_1_7/kernel

output_1_7/kernel/read

output_1_7/bias

output_1_7/bias/read

output_1_7/MatMul

output_1_7/BiasAdd

Есть ли способ узнать фактическую операцию ввода и вывода замороженного графа?


person tre95    schedule 25.04.2019    source источник


Ответы (1)


Хорошо для тех, кто ищет ответ: кажется, единственный надежный способ - визуализировать график с помощью Tensorboard. Таким образом вы сможете избежать любых ошибок, независимо от того, сколько операций содержит ваш график.

person tre95    schedule 29.04.2019