时间戳

首页 > 解决方案 > 如何从两个不同的 URL 获取/加载 model.json 和 weight.bin？

javascript - 如何从两个不同的 URL 获取/加载 model.json 和 weight.bin？

问题描述

我正在浏览保存和加载模型-Tensorflow文档页面以将模型加载/保存并运行到浏览器中。tensorflow.js 的 CDN 版本是1.1.2

例如使用的语法

const model = await tf.loadLayersModel('http://model-server.domain/download/model.json');

文档说：

这会从 http 端点加载模型。加载 json 文件后，该函数将请求 json 文件引用的相应 .bin 文件。

但是我正在使用的系统不允许两个不同的文件具有相同的 URL 路径，因为每个文件都保存有一个单独的 ID。

所以我的问题是，如何从两个不同的 URL 地址加载/保存 model.json 和 weight.bin（https 方法）？

顺便说一句，如果tf.loadFrozenModel (modelUrl, weightsManifestUrl, requestOption?)可以使用，那么请告诉我如何使用它？因为，它对我不起作用。

标签： javascripttensorflowservicenowtensorflow.js

解决方案

如果要从 2 个不同的位置加载模型和权重，根据文档，可以使用以下内容：

const model = await tf.loadLayersModel('http://model-server.domain/download/model.json', {weightPathPrefix: weigthsUrl});

推荐阅读