在tensorflow中的Python API中找不到某些功能

时间:2019-01-31 02:20:18

标签: tensorflow

我正在尝试将模型从tensorflow转换为onnx。这样做的过程如下。

  1. 为张量流中的权重保存一个graph_def和一个ckpt
  2. 检查graph_def的结构是否有效,并提供输入和输出是什么。
  3. 将它们一起冻结到frozen tensorflow graph中。
  4. 将该图转换为onnx模型。

问题出在步骤2中。要检查图形定义,我尝试在enter image description here中调用summary_graph。但是,它不能正常工作。接下来,我找到了Graph Transform Tool的文档。根据文档,他们使用bazel,它是一种类似于maven的构建和测试工具。这意味着我不能在从pip软件包管理器安装的tensorflow中使用此功能吗?使用此功能的唯一方法是从源代码安装张量流并使用bazel进行构建?

1 个答案:

答案 0 :(得分:0)

您应该完全能够使用这些功能从pip安装TensorFlow。 Bazel用于管理构建过程,除非您希望自己从源代码编译TensorFlow,否则不需要它。

尝试删除它并从pip重新安装,请注意选择正确的Python设置,以防您的计算机上有多个Python发行版。