javascript - 在javascript中从本地文件系统加载Tensorflow js模型

Question

我已经将一个 keras 模型转换为 tensorflow json 格式并将其保存在我的本地计算机中。我正在尝试使用以下命令在 javascript 代码中加载该 json 模型

model = await tf.loadModel('web_model')

但是模型没有加载。 有没有办法从本地文件系统加载 tensorflow json 模型？

Answer 1

我知道您正在尝试在浏览器中加载您的模型，但如果有人尝试在 Node 中加载您的模型，请按以下方式操作:

const tf = require("@tensorflow/tfjs");
const tfn = require("@tensorflow/tfjs-node");
const handler = tfn.io.fileSystem("./path/to/your/model.json");
const model = await tf.loadLayersModel(handler);