Java使用 tensorflow 图调用python函数

Question

所以我有一个 tensorflow (python2.7)中的神经网络，我需要使用Java检索它的输出。我有一个简单的 python 函数 getValue(input) ，它启动 session 并检索值。我愿意接受任何建议。我相信 Jython 不会工作，因为库中没有 tensorflow 。我需要尽快打电话。 JNI 存在用于 Java 调用 C，所以我可以使用 cython 进行转换并编译然后使用 JNI 吗？有没有办法在 RAM 中传递信息或者我没有想到的其他方式？

Answer 1

在 Python 中，保存模型(使用 saver.save)和图形(使用 tf.train.write_graph)。

在 Java 中，使用 org.bytedeco.javacpp-presets 库从保存的 protobuf 文件实例化 GraphDef，并传入输入特征并在 session 中获取输出特征。

参见https://medium.com/google-cloud/how-to-invoke-a-trained-tensorflow-model-from-java-programs-27ed5f4f502d#.4su1s26fz例如代码。