在 Node.js 中加载 TensorFlow 冻结图 (.pb)
Loading a TensorFlow frozen graph (.pb) in Node.js
我将一个tensorflow模型保存在一个冻结的PB文件中,适合TensorFlow Lite使用。
此文件可以在 Android 中加载,并通过以下代码运行良好:
import org.tensorflow.contrib.android.TensorFlowInferenceInterface;
…
TensorFlowInferenceInterface inferenceInterface;
inferenceInterface = new TensorFlowInferenceInterface(context.getAssets(), "MODEL_FILE.pb");
有什么方法可以加载Node.js中的冻结图吗?
我找到了解决方案 here:
1. 模型必须转换为网络友好格式,即 JSON 文件。
2.然后可以在Nodejs中使用'@tensorflow/tfjs'加载。
我将一个tensorflow模型保存在一个冻结的PB文件中,适合TensorFlow Lite使用。 此文件可以在 Android 中加载,并通过以下代码运行良好:
import org.tensorflow.contrib.android.TensorFlowInferenceInterface;
…
TensorFlowInferenceInterface inferenceInterface;
inferenceInterface = new TensorFlowInferenceInterface(context.getAssets(), "MODEL_FILE.pb");
有什么方法可以加载Node.js中的冻结图吗?
我找到了解决方案 here: 1. 模型必须转换为网络友好格式,即 JSON 文件。 2.然后可以在Nodejs中使用'@tensorflow/tfjs'加载。