在Node.js中使用Azure自定义视觉模型

Question

我对Tensorflow.js并不熟悉，所以我并不完全理解它。我使用Azure (customvision.ai)训练了一个对象检测模型，如文档中所述，我将该模型导出用于Tensorflow.js，以便在Tensorflow.js项目中脱机使用。

他们有一个关于Github (https://github.com/microsoft/customvision-tfjs)的非常简短的指南，介绍如何在Tensorflow.js中使用这个模型，我试着遵循这些步骤，但是无法运行它。

如果有人有Tensorflow.js的经验，请帮助我在这方面我是卡在过去3天。

这是我正在尝试执行的代码，我得到了错误

“等待只能与异步函数一起使用”

尽管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

您看到ONNX.js使用JavaScript运行ONNX模型了吗？https://github.com/microsoft/onnxjs

​

​