node.js - TFJS- Node : How to load model from url?

Question

我想从 Node 中的 url 加载模型。

这在兄弟中有效:

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

但不在 Node 中

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

我可以像这样通过 Node 获取请求:

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

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

Answer 1

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

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

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