在Node.js中加载TensorFlow冻结图（.pb）

Question

我将一个Tensorflow模型保存在一个冻结的PB文件中，该文件适合TensorFlow Lite使用。 此文件可以在Android中加载，并可以通过以下代码正常运行：

import org.tensorflow.contrib.android.TensorFlowInferenceInterface; … TensorFlowInferenceInterface inferenceInterface; inferenceInterface = new TensorFlowInferenceInterface(context.getAssets(), "MODEL_FILE.pb");

有什么方法可以在Node.js中加载冻结图吗？

Answer 1

我找到了解决方法here： 1.模型必须转换为Web友好格式，即JSON文件。 2.然后可以在Node.js中使用“ @ tensorflow / tfjs”进行加载。