博主为了移植好自己的tf模型到Movidius神经棒上,需要找到模型输入与输出的节点
找遍全网并没有实用的方法,最后在tensorflow官方自带的tool里找到了能看到输入与输出节点的代码
1.安装Bazel
首先要安装Bazel这个软件,没有这个是跑不起来的,可以去Bazel的官网查看安装步骤
在此我给出基于Ubuntu16.04下的安装方法
Step 1: 安装JDK
$ sudo apt-get install openjdk-8-jdk
Step 2: 添加Bazel源
echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -
Step 3: 更新Bazel的库
$ sudo apt-get update && sudo apt-get install bazel
$ sudo apt-get upgrade bazel
2. 下载tensorflow的自带的工具
下载地址 如果下载过慢可以修改host文件
3.运行指令
其实也就两句指令的事,第一句指令第一次运行需要等五分钟左右
bazel build tensorflow/tools/graph_transforms:summarize_graph
注意第二句指令中的 --in_graph= 后面需要更改为自己的,pb文件的位置
bazel-bin/tensorflow/tools/graph_transforms/summarize_graph --in_graph=tensorflow_inception_graph.pb
运行之后出现这样类似这样的界面就算是成功了
其中可以看到输入节点名称为 input
输出节点为 InceptionV2/Predictions/Reshape_1
有兴趣的话,可以看看我的tensorflow QA合集