如何在 Node.js 上保存的浏览器中使用 tensorflow.js 加载模型和权重？

Question

来源：Save / load model docs

我已经在 nodeJS 端训练了多个模型并将它们保存到使用 "file://"

所以我有 1 个 JSON 文件和 1 个带权重的二进制文件

但要在浏览器端加载此模型，我只能使用 localStorage、indexedDB 和 HTTP 请求。

加载它们的正确方法是什么？

我无法在将模型放在那里之前从 localStorage 加载。 也许我可以以某种方式将我的两个文件转换成类似 JSON 的形式并将其放入 localStorage?

浏览器重启后正在清理 IndexedDB。不知道能不能用

Http 似乎不错，但我如何使用加载权重？

await tf.loadModel('http://model-server.domain/download/model.json')

它只是一个文件请求。

有什么想法吗？

Answer 1

使用与模型文件相同的路径自动加载权重文件。 在您的示例中，模型文件具有 url 如下所示： http://model-server.domain/download/model.json

加载程序将从以下 url 加载权重文件： http://model-server.domain/download/group1-shard1of1

只要您将权重文件存储在服务器上与 model.json 文件相同的目录中，它就应该可以工作。