在没有安装 Tensorflow 的情况下运行 Tensorflow 模型

Question

我有一个运行良好的 TF 模型，它是用 Python 和 TFlearn 构建的。有没有办法在另一个系统上运行这个模型而不安装 Tensorflow？它已经经过预训练，所以我只需要通过它运行数据。

我知道 tfcompile （此处为Thread），但设置起来似乎相当复杂。有没有其他选择？

Answer 1

\n
\xc2\xa0有没有办法在另一个系统上运行这个模型而不需要安装Tensorflow？它已经经过预先训练，所以我只需要通过它运行数据即可。
\n

\n\n

是的

\n\n

训练完模型后。使用tf.python.tools.freeze_graph和tf.python.tools.optimize_for_inference_lib冻结和优化模型，以便在 Android 等其他设备上进行推理。

\n\n

上面的输出将是

\n\n

\n\n

[这些函数会将模型的所有变量转换为常量运算并导出到 protobuf 文件]

\n\n

使用优化的图 protobuf 文件并使用 Java 和其他 Tensorflow API 中可用的推理方法加载它。传递数据并获取输出。

\n\n

[请注意，您没有安装完整的 Tensorflow，但您只需要推理库]

\n\n

这里演示了一个简单的例子：

\n\n

\n\n

它适用于 Android，但过程对于 Java 应该是相同的。

\n\n

对于 C++：单击此处

\n