试图使用来自coco-ssd的数据来绘制一个边界框。

Question

我最近一直在询问关于图像识别的问题，并发现JavaScript有一个简单易用且对初学者友好的方法，使用tensorflow.js来实现这个目标。我设法将示例代码用于我发现的一个有趣的模型，叫做coco-ssd。它工作得非常好，并返回预测的数据（它返回一个包围主题的四个点的数组）和一切，但现在我有使用它的问题。为了达到我想要的目的，我希望能够在帧中的人周围画一个边界框，或者至少知道原始图像的像素坐标（调整大小一直是一个真正的问题）。我不知道该怎么做，因为我所看到的所有教程都是使用了 <canvas> 元素。当我尝试使用这个功能时，画布似乎永远无法正常显示。下面是代码（为了方便，我把代码从一个单独的文件移到了脚本标签中）。

<script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs"> </script>

<script src="https://cdn.jsdelivr.net/npm/@tensorflow-models/coco-ssd"> </script>
<!-- replace 'Lenin.jpg' with a local file in the same folder as this program -->
<img id="img" src="Lenin.jpg" crossorigin="anonymous"/>

<script>


  const img = document.getElementById('img');

  cocoSsd.load().then(model => {
    model.detect(img).then(predictions => {
      console.log('Predictions: ', predictions);
    });
  });
</script>

任何帮助将是感激的！

Answer 1

请看我的COCO-SSD入门教程，它展示了如何为单幅图像和网络摄像头做这件事。

https:/glitch.com~tensorflow-js-object-detection(对象检测)