将模型转换为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

您没有提供完整的代码，但让我猜测，您有一个如下所示的import语句：

from tensorflow.tools import freeze_graph

应该是：

from tensorflow.tools.freeze_graph import freeze_graph

这是因为freeze_graph函数实际上也存在于称为freeze_graph的模块中。您使用的import语句将导入整个模块。准确地说，当您的代码到达freeze_graph部分时，就会出现错误，指出模块不可调用：）