node.js - 将 Azure 自定义视觉模型与 Node.js 结合使用

Question

我是 Tensorflow.js 的新手，所以我不太了解它。我已经使用 Azure 自定义视觉 ( customvision.ai ) 训练了一个对象检测模型，并且如文档中所述，我导出了 Tensorflow.js 的模型，以便在我的 Node.js 项目中离线使用。

他们在 Github ( https://github.com/microsoft/customvision-tfjs ) 上有一个关于如何将模型与 Tensorflow.js 一起使用的非常简短的指南，我尝试按照这些步骤操作，但无法运行它。

如果有人对 Tensorflow.js 有经验，请帮助我，我在过去 3 天里一直坚持下去。

这是我要执行的代码，我收到错误

'await 只能与异步函数一起使用'

即使 loadModelAsync() 函数是异步的。

const cvstfjs = require('@microsoft/customvision-tfjs');

let model = new cvstfjs.ObjectDetectionModel();
await model.loadModelAsync('model.json');
const image = document.getElementById('image');
const result = await model.executeAsync(image);

Answer 1

正如错误所说，您需要将任何await调用包装在一个async函数中：

const cvstfjs = require('@microsoft/customvision-tfjs');

async function doThings() {
  let model = new cvstfjs.ObjectDetectionModel();
  await model.loadModelAsync('model.json');
  const image = document.getElementById('image');
  const result = await model.executeAsync(image);
  return result;
}

doThings().then((result) => {
  console.log(result);
});

Answer 2

您是否看到使用 JavaScript 运行 ONNX 模型的 ONNX.js。 https://github.com/microsoft/onnxjs