.pbを解凍して全ノード名を記述したテキストファイル生成

.pbを解凍して全ノード名を書き込む

今回は, どちらかというと備忘録です.
個人的にNetworkの全ノード名を知りたかったので, テキストファイルに記述するような関数を作成しました.

writeTensorsName(pb file Path, Output text file Path)
→ Networkの全ノード名が記述されたText file

# .pbファイルをロードしてテキストファイルにネットワークの名前を書き込む
def writeTensorsName(pb_file, txt_file):
    # read pb into graph_def
    with tf.gfile.GFile(pb_file, "rb") as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())

    # import graph_def
    with tf.Graph().as_default() as graph:
        tf.import_graph_def(graph_def)

    # print operations
    path = pathlib.Path(txt_file)
    with path.open(mode='w') as f:
        # for op in graph.get_operations():
        #     f.write(op.name+'\n')

       for op in graph.get_operations():
           for t in op.values():
                f.write(t.name+'\n')

writeTensorsName('model.pb', 'model_name.txt')