如何查找冻结模型的输入和输出节点

pip install tensorflow

saved_model_cli show --dir /path/to/saved_model_directory --all

import tensorflow as tf
from tensorflow.core.protobuf import saved_model_pb2
from tensorflow.python.platform import gfile
from tensorflow.python.summary import summary

# 载入模型
with tf.Session() as sess:
    model_filename ='/path/to/your/model.pb'
    with gfile.FastGFile(model_filename, 'rb') as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())
        sess.graph.as_default()
        tf.import_graph_def(graph_def, name='')

    # 将模型写入TensorBoard
    train_writer = tf.summary.FileWriter('/path/to/logdir')
    train_writer.add_graph(sess.graph)

tensorboard --logdir=/path/to/logdir

import tensorflow as tf

# 载入模型
model_filename = '/path/to/your/model.pb'
with tf.gfile.GFile(model_filename, "rb") as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())

# 导入图
with tf.Graph().as_default() as graph:
    tf.import_graph_def(graph_def, name="")

    # 打印操作名称
    for op in graph.get_operations():
        print(op.name)  # 打印操作的名字
        print(op.values())  # 打印张量名称和尺寸