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

时间:2020-02-17 12:19:10

标签: node.js tensorflow tensorflow.js azure-cognitive-services microsoft-custom-vision

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

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

如果任何人都有Tensorflow.js的经验,请帮助我,在过去的3天里我都被困在上面。

这是我要执行的代码,出现错误

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

即使loadModelAsync()函数是Async。

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);

2 个答案:

答案 0 :(得分: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);
});

答案 1 :(得分:0)

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

enter image description here