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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值