tensorboard常用API总结

1、tf.summary.image(name, tensor, max_outputs=3, collections=None, family=None)
可视化当前使用的训练、测试图片

参数：
name：生成的节点的名称。也将作为TensorBoard中的系列名称。
tensor：uint8或者float32型的4-D Tensor[batch_size, height, width, channels]，其中channels为1， 3 或者4。
max_outpus：要生成图像的最大批处理元素数。
collections：ops.GraphKeys的可选列表。要添加摘要的集合。默认为[_ops.GraphKeys.SUMMARIES]。
family：可选的; 如果提供，则用作摘要标记名称的前缀，该名称控制用于在Tensorboard上显示的选项卡名称。

tf.summary.image('input', image_shaped_input, 10)

2、tf.summary.scalar(tags, values, collections=None, name=None)
监控标量数据随迭代进行的变化趋势。一般是cross_entropy、loss、accurary等

tf.summary.scalar('accuracy', accuracy)

3、tf.summary.merge_all()
将所有summary全部保存到磁盘，以便tensorboard显示

merged = tf.summary.merge_all()

4、 tf.summary.FileWriter(path,sess.graph)
生成一个写日志的writer，并将当前的计算图写入日志

summary_writer = tf.summary.FileWriter(SUMMARY_DIR, sess.graph)

5、summary_writer.add_summary(merged，step)
调用summary_writer的add_summary方法将训练过程以及训练步数保存

        for i in range(TRAIN_STEPS):
            xs, ys = mnist.train.next_batch(BATCH_SIZE)
            # 运行训练步骤以及所有的日志生成操作，得到这次运行的日志。
            summary, _ = sess.run([merged, train_step], feed_dict={x: xs, y_: ys})
            # 将得到的所有日志写入日志文件，这样TensorBoard程序就可以拿到这次运行所对应的
            # 运行信息。
            # merged为所有summary，i为step
            summary_writer.add_summary(summary, i)

6、tf.summary.histogram(tags, values, collections=None, name=None)
用来显示直方图信息，一般用来显示训练过程中变量的分布情况

tf.summary.histogram(‘weights’, var)

参考资料：
https://blog.csdn.net/baymax_007/article/details/85249432

tensorboard常用API总结

猜你喜欢