tensorboard查看pb
时间: 2023-11-06 09:57:03 浏览: 202
要查看pb文件的tensorboard可视化,可以按照以下步骤进行操作:
1. 在pb模型文件所在文件夹中新建一个Python文件,命名为check_pb.py。
2. 在check_pb.py文件中导入必要的库和模块,代码如下:
```
from tensorflow.python.platform import gfile
import tensorflow as tf
```
3. 使用tf.get_default_graph()获取默认的计算图,并将pb模型文件加载到计算图中,代码如下:
```
model = 'D:/test/model.pb'
graph = tf.get_default_graph()
graph_def = graph.as_graph_def()
graph_def.ParseFromString(gfile.FastGFile(model, 'rb').read())
tf.import_graph_def(graph_def, name='graph')
```
4. 创建一个tf.summary.FileWriter对象来保存图的摘要数据,代码如下:
```
summaryWriter = tf.summary.FileWriter('D:/test/log/', graph)
```
5. 运行check_pb.py文件来生成日志文件,代码如下:
```
python check_pb.py
```
6. 打开命令行,进入到check_pb.py文件所在的目录,运行以下命令来启动tensorboard:
```
tensorboard --logdir=D:/test/log/
```
7. 在浏览器中打开生成的链接,即可在tensorboard中查看pb文件的可视化结果。
阅读全文