Найти параметры графика Tensorflow - PullRequest
0 голосов
/ 27 марта 2019

У меня есть этот график TensorFlow. Я успешно получил input_arrays и output_arrays этого графика, используя этот скрипт:

import tensorflow as tf
gf = tf.GraphDef()   
m_file = open('deeplabv3_mnv2_pascal_tain.pb','rb')
gf.ParseFromString(m_file.read())

for n in gf.node:
    print( n.name )

Я застрял в точке, где я не могу получить:

input_shapes

из того же графика. Как я могу получить эту информацию, если у меня есть только этот график.

...