如何将预训练的tensorflow模型加载并预测到Java代码中？

Question

我不确定将预先训练好的keras模型加载到Java代码中的常见做法是什么？在

我看到了deeplearning4j和{a2}。在

令人惊讶的是，谷歌说：

Caution: The TensorFlow Java API is not covered by the TensorFlow API stability guarantees.

我真的很困惑，在选择tensorflow Java引擎之前，这里发生了什么，我需要考虑什么。显然，我想支持最新的CUDA/cuDNN。我的应用程序运行在个人电脑上，不是移动应用程序。在

Answer 1

您可以按照以下步骤操作：

1. 您可以使用TensorFlow Lite或TensorFlow Mobile。建议使用TensorFlow Lite。访问here。在
2. TensorFlowLite有一个gradle和maven依赖，这在Android和java中很有用。在
3. 要将python模型转换为TF-Lite文件，请使用notebook中的函数
4. 它可以在您用python训练的模型上运行推断。但是，TF-Lite存在LSTM单元和批处理规范化层的问题。在