加载pb模型文件,并输出定义
model = 'model.pb'
with tf.Session() as sess:
with open(model, 'rb') as model_file:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
print(graph_def)
采用上述的方式可以在新的会话中重新加载本地的模型文件(pb),然后二进制解析后,输出可以看到结果。但是如果网络层结构十分复杂,那么这种显示方式就会比较难以阅读。
加载pb模型文件,并使用Tensorboard可视化
import tensorflow