码迷,mamicode.com
首页 > 其他好文 > 详细

使用TensorBoard

时间:2020-05-18 22:18:10      阅读:58      评论:0      收藏:0      [点我收藏+]

标签:打开   pes   输出   span   color   rip   orb   style   class   

偶然机会需要使用ckpt转换为pb,所以需要查找到输出节点。

 

指定ckpt文件,然后会在当前文件夹下生成__tb文件夹,里面还有一个文件。

import tensorflow as tf


sess = tf.Session()
tf.train.import_meta_graph("F:/ModelArts/cityscapesScripts/exp/train_on_train_set/train/model.ckpt-1000.meta")
tf.summary.FileWriter("__tb", sess.graph)

切换到指定目录

技术图片

 

 最后会有一个链接

技术图片

 

 在浏览器中复制打开

技术图片

 

使用TensorBoard

标签:打开   pes   输出   span   color   rip   orb   style   class   

原文地址:https://www.cnblogs.com/birdmmxx/p/12913223.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!