Question

我正在尝试将我的* .pb tensorflow模型转换为coreML。我一直在坚持识别模型的输出节点。

为了获得我的输出节点，我试图在我的* .pb文件上构建并运行summarize_graph，但遇到了问题。下载源代码后如何构建和运行summarize_graph？

我运行以下命令：

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

我收到以下错误：

信息：分析了0个目标（加载了0个包）。信息：找到0个目标...... 信息：经过时间：0.389秒，关键路径：0.01秒信息：构建完成成功，共1次行动 bazel-bin / tensorflow / tools / graph_transforms / summarize_graph：没有这样的文件或目录

发出bazel命令后，我执行命令的位置会出现一个空白的bazel-bin目录。

注意，我的tensorflow安装中不存在summarize_graph。所以我从github tensorflow / tools / graph_transforms下载了源码并将其复制到我的tensorflow / tools / graph_transforms目录中。

该目录包含以下内容：

构建README.md 的初始化的.py init .pyc add_default_attributes.cc add_default_attributes_test.cc backports.cc backports_test.cc compare_graphs.cc fake_quantize_training.cc fake_quantize_training_test.cc file_utils.cc file_utils.h file_utils_test.cc flatten_atrous.cc flatten_atrous_test.cc fold_batch_norms.cc fold_batch_norms_test.cc fold_constants_lib.cc fold_constants_lib.h fold_constants_test.cc fold_old_batch_norms.cc fold_old_batch_norms_test.cc freeze_requantization_ranges.cc freeze_requantization_ranges_test.cc fuse_convolutions.cc fuse_convolutions_test.cc insert_logging.cc insert_logging_test.cc obfuscate_names.cc obfuscate_names_test.cc out python quantize_nodes.cc quantize_nodes_test.cc quantize_weights.cc quantize_weights_test.cc remove_attribute.cc remove_attribute_test.cc remove_device.cc remove_device_test.cc remove_ema.cc remove_ema_test.cc remove_nodes.cc remove_nodes_test.cc rename_attribute.cc rename_attribute_test.cc rename_op.cc rename_op_test.cc round_weights.cc round_weights_test.cc set_device.cc set_device_test.cc sort_by_execution_order.cc sort_by_execution_order_test.cc sparsify_gather.cc sparsify_gather_test.cc strip_unused_nodes.cc strip_unused_nodes_test.cc summarize_graph_main.cc transform_graph.cc transform_graph.h transform_graph_main.cc transform_graph_test.cc transform_utils.cc transform_utils.h transform_utils_test.cc

我在macbook pro上

谢谢！

Answer 1

如果有人遇到类似的问题，我就解决了。

cmd> ./configure
cmd> bazel build tensorflow/tools/graph_transforms:summarize_graph

（您可能会收到有关xcode的错误;如果是，请运行以下命令）

cmd> xcode-select -s /Applications/Xcode.app/Contents/Developer

cmd> bazel clean --expunge

cmd> bazel build tensorflow/tools/graph_transforms:summarize_graph

Answer 2

CentOS 7演练：

yum install epel-release
yum update
yum install patch
curl https://copr.fedorainfracloud.org/coprs/vbatts/bazel/repo/epel-7/vbatts-bazel-epel-7.repo -o /etc/yum.repos.d/vbatts-bazel-epel-7.repo
yum install bazel
curl -L -O https://github.com/tensorflow/tensorflow/archive/v1.8.0.tar.gz
cd tensorflow-1.8.0
./configure  # interactive!
bazel build tensorflow/tools/graph_transforms:summarize_graph
bazel-bin/tensorflow/tools/graph_transforms/summarize_graph

tensorflow工具：无法弄清楚如何构建/运行summarize_graph

2 个答案: