我将 tensorflow 模型保存在适合 TensorFlow Lite 使用的卡住 PB 文件中。 该文件可以在 Android 中加载,并且可以通过以下代码正常运行:
导入 org.tensorflow.contrib.android.TensorFlowInferenceInterface;
……
TensorFlowInferenceInterface 推理接口(interface);
inferenceInterface = new TensorFlowInferenceInterface(context.getAssets(), "MODEL_FILE.pb");
有没有办法在 Node.js 中加载卡住图?
最佳答案
我找到了解决方案here : 1. 模型必须转换为 Web 友好格式,即 JSON 文件。 2.然后可以在Nodejs中使用'@tensorflow/tfjs'加载。
关于node.js - 在 Node.js 中加载 TensorFlow 卡住图 (.pb),我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/54026922/