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

Question

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

这在broser中有效：

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

但不在节点中

  

错误：没有获取polyfill的网络浏览器不支持browserHTTPRequest

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

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

get请求返回一个具有modelTopology和weightManifest的Object。我该如何从中创建模型？

Answer 1

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

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

tf.loadModel在后​​台使用fetch。但是fetch在nodeJs中本身不受支持。这就是为什么应该输入pollyfill的原因。