Tensorflow之pb文件分析
使用Tensorboard分析pb文件有两种方法:
方法一:
利用pb文件恢复计算图
利用Tensorboard查看计算图的结构
方法二
- 利用tensorflow提供的tools里的import_pb_to_tensorboard.py这个工具,但是这个工具linux版本的tensorflow没有安装(Win下默认安装),需要的可以去下载[https://github.com/tensorflow/tensorflow/tree/master/tensorflow/python/tools]
方法一
1.从pb文件中恢复计算图
import tensorflow as tfmodel = 'model.pb' #请将这里的pb文件路径改为自己的
graph = tf.get_default_graph()
graph_def = graph.as_graph_def()
graph_def.ParseFromString(tf.gfile.FastGFile(model, 'rb').read())
tf.import_graph_def(graph_def, name='graph')
summaryWriter = tf.summary.FileWriter('log/', graph)
2.利用Tensorboard查看计算图
在命令行运行以下命令,启动Tensorboard
#命令行运行里执行
tensorboard --logdir log/ #这里的路径就是1中最后一行图保存的路径,请根据自己的需要更改
方法二
利用tools里面的import_pb_to_tensorboard.py工具
#命令行
python -m tensorflow.python.tools.import_pb_to_tensorboard \--model_dir="your_path/model.pb" --log_dir="your_log_path"
tensorboard --logdir="your_log_path" #启动tensorboard
或者
#python3
from tensorflow.python.tools.import_pb_to_tensorboard import import_to_tensorboard
model = os.path.join(model_dir, 'tensorflow_inception_graph.pb')
import_to_tensorboard(model_dir=model, log_dir='log/')
#命令行
tensorboard --logdir="your_log_path" #启动tensorboard
经过查看源码,第二种方法其实是对第一种方法的包装。
两种方法是一致的,只不过第二种方法更加便捷。
参考资料:
- https://blog.csdn.net/szfhy/article/details/79567135
- https://blog.csdn.net/twt520ly/article/details/80228970
Tensorflow之pb文件分析相关推荐
- tensorflow查看pb文件
运行下面代码: from tensorflow.python.platform import gfile import tensorflow as tfmodel = './saved_pb/expe ...
- tensorflow之pb文件保存与载入
pb是protocol(协议) buffer(缓冲)的缩写.TensorFlow训练模型后存成的pb文件,是一种表示模型(神经网络)结构的二进制文件,将图中的变量保存成为常量,便于调用,一般无法将pb ...
- TensorFlow:将ckpt文件固化成pb文件
TensorFlow:将ckpt文件固化成pb文件 本文是将yolo3目标检测框架训练出来的ckpt文件固化成pb文件,主要利用了GitHub上的该项目. 为什么要最终生成pb文件呢?简单来说就是直接 ...
- TF之TFOD-API:基于tensorflow框架利用TFOD-API脚本文件将YoloV3训练好的.ckpt模型文件转换为推理时采用的.pb文件
TF之TFOD-API:基于tensorflow框架利用TFOD-API脚本文件将YoloV3训练好的.ckpt模型文件转换为推理时采用的frozen_inference_graph.pb文件 目录 ...
- 使用C#把Tensorflow训练的.pb文件用在生产环境
训练了很久的Tf模型,终于要到生产环境中去考验一番了.今天花费了一些时间去研究tf的模型如何在生产环境中去使用.大概整理了这些方法. 继续使用分步骤保存了的ckpt文件 这个貌似脱离不了tensorf ...
- tensorflow保存数据为.pb格式和加载.pb文件(转)
最近接触了tensorflow的object detection API发现里面读取的预先训练模型都是pb格式. 谷歌推荐的保存模型的方式是保存模型为 PB 文件,它具有语言独立性,可独立运行,封闭的 ...
- VS2015+OpenCV3.4.5+QT5.12+WINDOWS10用c++调用tensorflow训练好的.pb文件图像检测
版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明. 本文链接:https://blog.csdn.net/qq_31806049/article/ ...
- 【tensorflow】生成.pb文件
Saving, Freezing, Optimizing for inference, Restoring of tensorflow models 在训练完tensorflow模型后,会有三个文件: ...
- tensorflow | tensorflow实现将ckpt转pb文件
本博客实现将自己训练保存的ckpt模型转换为pb文件,该方法适用于任何ckpt模型,当然你需要确定ckpt模型输入/输出的节点名称. 目录 tensorflow实现将ckpt转pb文件 一.CKPT ...
最新文章
- python数据结构与算法:二分查找
- Docker 宿主机定时清除容器的运行日志
- python将一个json字典转换为一个python对象
- c语言分手代码大全,C语言代码大全
- go标准库的学习-net/rpc/jsonrpc
- DHCP 服务原理:Snooping和Relay
- extjs插件开发上传下载文件简单案例
- .NET Core 中如何调试 死锁 ?
- jQuery页面滚动 动态加载图片等元素
- 从此使用linux系统,但是QQ是必不可少的!!该篇文章方法成功!!!已验证!!!!!...
- 怎样测网络服务器延迟,网速测试延迟(网络延迟多少正常)
- Acer 常见笔记本产品内存扩展对照表
- iOS 谷歌地图及路线规划
- python爬虫获取维基百科词条
- php实例精通txt下载,PHP实例精通(1碟)
- 慢下来,等一等自己的灵魂
- 微信图片太模糊如何弄清晰?微信图片过期怎么恢复原图
- chipseq MACS2 call peaks 报错解决方法——创建虚拟环境
- linux 打开网页 看不清网页上的字体呈现出框,电脑浏览器打开网页字体显示模糊如何解决...
- IIC总线协议---以存储芯片at24c64为例