如何在Tensorflow中用bazel编译summarize_graph

Question

我在测试一个运动神经计算棒。我的第一个测试是用Inception_v4模型：modelzoo.html复制本指南

我确实需要知道freeze_graph.py的“freeze_graph.py”。我在tensorflow docs中找到了这样做的指令：

bazel build tensorflow/tools/graph_transforms:summarize_graph
bazel-bin/tensorflow/tools/graph_transforms/summarize_graph --in_graph=MY_PB_FILE.pb

但是，在尝试执行步骤时，我会得到以下错误：

lberto@alberto-OptiPlex-9010:~/movidius_ws$ bazel build  tensorflow/tools/graph_transforms:summarize_graph
Starting local Bazel server and connecting to it...
ERROR: Skipping 'tensorflow/tools/graph_transforms:summarize_graph': no such package 'tensorflow/tools/graph_transforms': BUILD file not found on package path
WARNING: Target pattern parsing failed.
ERROR: no such package 'tensorflow/tools/graph_transforms': BUILD file not found on package path
INFO: Elapsed time: 0.947s
INFO: 0 processes.
FAILED: Build did NOT complete successfully (0 packages loaded)

根据Movidius提供的指南，没有必要在主机上安装tensorflow，只需下载源代码。我在目录中克隆了Tensorflow源代码以遵循指南，movidius工具在共享该目录的docker容器中复制。

我在构建summarize_graph应用程序时做错了什么？我需要运行这个应用程序来了解我想要与movidius神经计算棒一起使用的output_node_names。

谢谢。

Answer 1

确保存在以下文件：

• “~/movidius_ws/工作区”
• "~/movidius_ws/tensorflow/tools/graph_transforms/BUILD“或"~/movidius_ws/tensorflow/tools/graph_transforms/BUILD.bazel”

如果有，请检查“//tensorflow/tools/”包中的目标列表：

bazel query //tensorflow/tools/graph_transforms:*

看看"summarize_graph“是否存在。