我无法使用 @tensorflow/tfjs 从 file:// 加载模型

Question

我正在构建一个 npm 包，用于加载预训练的 tensorflow 模型并进行一些预测。

当我用tf = require('@tensorflow/tfjs-node') 做tf.loadLayersModel() 时，我没有问题。不幸的是，该库在安装时似乎有问题。 我尝试改用tf = require('@tensorflow/tfjs)。虽然，我收到以下错误：

UnhandledPromiseRejectionWarning: E​​rror: browserHTTPRequest 在没有 fetch polyfill 的情况下不受 Web 浏览器外部支持。

这是我的代码：

 // Load ML model
  tf.loadLayersModel("file://" + __dirname + "mymodel/model.json").then(
    model => {
      }
    }
  );

有办法解决这个问题吗？

Answer 1

在使用 Node.js 时最好使用 @tensorflow/tfjs-node。 只是不需要@tensorflow/tfjs，一切都会按预期进行。

但如果你真的只想使用@tensorflow/tfjs，那么只需添加tfjs-node-save

npm install --save tfjs-node-save

并在你的脚本中使用它

require('tfjs-node-save');

Answer 2

UnhandledPromiseRejectionWarning: E​​rror: browserHTTPRequest 在没有 fetch polyfill 的情况下不受 Web 浏览器外部支持。

如果您使用的是 @tensorflow/tfjs 而不是 @tensorflow/tfjs-node，则需要使用 polyfill 进行获取

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