如何在Tensorflow中使用bazel编译summary_graph

时间:2018-11-14 13:43:53

标签: tensorflow bazel

我正在测试Movidius神经计算棒。 我的第一个测试是使用Inception_v4模型复制此指南: https://movidius.github.io/ncsdk/tf_modelzoo.html

我确实需要知道Frozen_graph.py的“ output_node_name”。 我在tensorflow文档中找到了这样做的说明:

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源代码克隆到一个目录中以遵循该指南,并在共享该目录的docker容器中克隆了movidius工具。

构建summary_graph应用程序时我做错了什么? 我需要运行此应用程序才能知道要与movidius神经计算棒一起使用的output_node_names

谢谢。

1 个答案:

答案 0 :(得分:0)

确保存在以下文件:

  • “〜/ movidius_ws / WORKSPACE”
  • “〜/ movidius_ws / tensorflow / tools / graph_transforms / BUILD”或“〜/ movidius_ws / tensorflow / tools / graph_transforms / BUILD.bazel”

如果这样做,请检查“ // tensorflow / tools / graph_transforms”包中的目标列表:

bazel query //tensorflow/tools/graph_transforms:*

看看是否有“ summarize_graph”。