在 Node.js 中加载 TensorFlow 冻结图 (.pb)答案

【问题标题】：Loading a TensorFlow frozen graph (.pb) in Node.js在 Node.js 中加载 TensorFlow 冻结图 (.pb)
【发布时间】：2019-01-04 02:19:17
【问题描述】：

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

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

有没有办法在 Node.js 中加载冻结图？

【问题讨论】：

标签： node.js tensorflow tensorflow-lite

【解决方案1】：

我找到了解决方案here： 1. 模型必须转换为 JSON 文件的网络友好格式。 2. 然后可以在Nodejs中使用'@tensorflow/tfjs'来加载。

【讨论】：