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

时间:2019-01-18 11:31:14

标签: tensorflow neural-network tensorflow.js

来源: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')

它只是一个文件请求。

有什么想法吗?

1 个答案:

答案 0 :(得分:1)

将使用与模型文件相同的路径自动加载重量文件。 在您的示例中,模型文件的网址如下: http://model-server.domain/download/model.json

加载器将从以下URL加载重量文件: http://model-server.domain/download/group1-shard1of1

只要将权重文件与服务器上的model.json文件存储在同一目录下,它就可以工作。