采用tensorflow自带工具查看输入与输出节点

博主为了移植好自己的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合集