tensorflow保存模型，使用tensorboard查看结点名称

有时候想看一下分类的置信度，需要知道tensor的名字，也就需要将模型可视化

 

保存模型

saver = tf.train.Saver()
saver.save(sess, "D:/all_model/model_7/model.ckpt")

然后转化为tensorboard可以识别的文件

import tensorflow as tf
from tensorflow.python.platform import gfile
graph = tf.get_default_graph()
graphdef = graph.as_graph_def()
_ = tf.train.import_meta_graph("D:/text_model/model.ckpt.meta")
summary_write = tf.summary.FileWriter("D:/all_model/tensorboard/" , graph)

然后就可以使用tensorboard来查看模型的每一个tensor的名字

tensorboard --logdir=D:/all_model/tensorboard

上述命令没有引号

查看模型

首先去查看置信度的tensor

这个是损失函数的选择，模型中并未命名，则默认名称是Softmax

所以查看置信度的时候直接get_tensor_by_name("Softmax:0")就可以了