从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的功能远不止这些,训练过程中的一些中间状态都可以查看。