我正在尝试将我的 *.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
谢谢!