TFJS-Node：如何从 url 加载模型？

Question

我想从节点中的 url 加载模型。

这在兄弟中有效：

mobileNet = await tf.loadModel('https://storage.googleapis.com/tfjs-models/tfjs/mobilenet_v1_0.25_224/model.json');

但不在节点中

Error: browserHTTPRequest is not supported outside the web browser without a fetch polyfill

我可以通过这样的节点获取请求：

const https = require('https');
https.get(mobileNetUrl, (res) => {
    ...
});

get 请求 returns 一个带有 modelTopology 和 weightManifest 的对象。我如何从中创建模型？

Answer 1

您需要将此行添加到您的代码中

global.fetch = require('node-fetch');

tf.loadModel 在后台使用 fetch。但是 fetch 在 nodeJs 中不是原生支持的。这就是应该导入 pollyfill 的原因。