如何快速查看tensorflow 的model文件结构

从github经常下载别人训练好的model,一个pb文件,里面都存储了什么东西呢?不像caffe,网络结果直接是一个文本文件,可以快速查看。

当然我们可以去找对应的文章,去查询网络结构等信息。但是tensorflow有更好的方法。

Tensorboard去解析这些model文件,很强大。

安装,pip install tensorboard

推荐使用whL的方式安装,当你机器上有多个tensorflow版本时,更应该使用whl方式来安装了。

使用anaconda方式安装的tensorflow,不同的机器上,安装的tools不一样,windows默认就安装了python/tools/import_pb_to_tensorboard.py这个工具,ubuntu默认就没有安装。

在终端上执行:

python tensorflow/python/tools/import_pb_to_tensorboard.py --model_dir="your_path/model.pb" --log_dir="your_log_path"
tensorboard --logdir="your_log_path" #启动了服务

在浏览器上输入:http://localhost:6006即可以图形化的方式查看model文件。

tensorboard的功能远不止这些,训练过程中的一些中间状态都可以查看。


评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值