将模型转换为 android 的 pb 文件时出错

Question

我收到此错误：

frozen_graph = freeze_graph(K.get_session(), output_names=[model.output.op.name])

TypeError: 'module' object is not callable

我正在使用 tensorflow 1.15.0 版和 keras 2.0.5

Answer 1

您没有包含完整的代码，但让我猜猜，您有一个如下所示的导入语句：

from tensorflow.tools import freeze_graph

它应该是：

from tensorflow.tools.freeze_graph import freeze_graph

这是因为freeze_graph函数实际上也存在于一个名为freeze_graph的模块中。 您使用的导入语句导入整个模块。 当您的代码到达freeze_graph部分时，非常准确，错误提示模块不可调用:)