【问题标题】:Java API for loading TensorFlow models created in python用于加载在 python 中创建的 TensorFlow 模型的 Java API
【发布时间】:2020-02-12 03:59:57
【问题描述】:
我正在使用 Java 13(我不确定这是否相关)并且我正在制作二十一点游戏。我想在我的 Java 应用程序中用 python 训练后添加一个神经网络模型。但是,在 here 上它说
“注意:目前还没有对 TensorFlow 2 的 libtensorflow 支持。”
我什至还没有开始制作我的模型,所以我没有尝试将模型加载到 Java 中。这会是个问题吗?我需要使用旧版本的 TensorFlow 吗?
【问题讨论】:
-
要在 Java 中运行 TF 2.0 模型,您可以改用 TensorFlow Lite。您可以使用 Gradle 构建它。见here。
-
哇,谢谢!在短暂阅读here@ShubhamPanchal 之后,它看起来也会有不错的表现
标签:
java
python
tensorflow
installation
java-13
【解决方案1】:
如果您不想将模型转换为 TF Lite,您还可以使用启用了 2.x 支持的 TensorFlow Java 可用的快照,请查看此repository。