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

Question

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

model = await tf.loadModel('web_model')

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

标签： javascripttensorflowtensorflow.js

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);