上文中提到:只有一个模型文件,怎么能知道模型结构呢?
想来想去只有使用tensorboard了。
那么直接使用下列代码:
saver = tf.train.import_meta_graph("./Model/ModelSoftmax.ckpt.meta")
saver.restore(sess, "./Model/ModelSoftmax.ckpt") # 注意此处路径前添加"./"
print('Load Model OK!')
print('ACC=',sess.run("Mean_1:0", feed_dict={"Placeholder:0":mnist.test.images,"Placeholder_1:0": mnist.test.labels}))
writer = tf.summary.FileWriter("D://TensorBoard//test",sess.graph)
这样就将graph写入到指定地址了。
然后启动命令行,输入命令:tensorboard –logdir=D://TensorBoard//test
就输出一个地址,然后打开google浏览器(千万注意,很多浏览器不可以,google的绝对可以),输入地址:
这个图简直太乱了,最简单的网络都乱成这样,可想而知,复杂的例如inceptionv4会是什么情况。所以这就扯出下一个问题:Tensor命名
仔细对了半天,最后才看出来训练时的y的tensor名字应该是:add:0
可以看到,预测输出不需要那个lable输入
see = sess.run("add:0", feed_dict={"Placeholder:0":mnist.test.images})
print(see.shape)
print('see=',see)