نمایش لایه ورودی و خروجی مدل(تنسورفلو)

Question

وقتی فایل مدل فریز شده داریم چطور تو کد نام لایه ورودی و خروجی را پیدا کنم؟

Answer 1

فایل graphdef را باز کنید و از طریق node می تونید به لایه های دسترسی داشته باشید.

    graph =  'model name'

    with tf.gfile.FastGFile(graph,'rb') as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        print(  graph_def.node[0])
        print(    graph_def.node[-1])