我正在测试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
谢谢。
答案 0 :(得分:0)
确保存在以下文件:
如果这样做,请检查“ // tensorflow / tools / graph_transforms”包中的目标列表:
bazel query //tensorflow/tools/graph_transforms:*
看看是否有“ summarize_graph”。