6

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

为了获得我的输出节点,我尝试在我的 *.pb 文件上构建和运行 summarise_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 目录。

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

该目录包含以下内容:

构建 README.md 初始化.py 初始化.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 fold_constants_lib.cc 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 .cc transform_graph.cc transform_graph.h transform_graph_main.cc transform_graph_test.cc transform_utils.cc transform_utils.h transform_utils_test.cccc 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.cccc 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

谢谢!

4

2 回答 2

4

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

导航到 tensorflow 源目录的根目录

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
于 2018-01-13T01:58:59.250 回答
1

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
于 2018-06-14T10:16:44.287 回答