使用TensorFlow.js进行实时单个对象跟踪

Question

就像每个人一样，我正在迈出机器学习的第一步，已经经历了许多TensorFlow.js教程，并且我正在尝试实现这一目标：“实时单个对象跟踪/检测”

类似这样的东西->输入：摄像头/视频->输出：对象边界框

我知道有SSD和YOLO，以及其他用于预测和定位对象的库。但是，预测时间非常慢（在浏览器中），我猜这是因为神经网络必须预测这么多对象之间。

https://github.com/ModelDepot/tfjs-yolo-tiny https://github.com/tensorflow/models/tree/master/research/object_detection

如果我只想跟踪单个对象怎么办？这有没有可能？性能会更好吗？我应该从哪里开始？

我一直在考虑从SavedModel中提取经过预训练的类（对象），然后开始从中进行更多训练。但是Google似乎没有任何指示：）

如果有人可以提供任何线索，我将不胜感激！

感谢进阶！

Answer 1

我在IBM的推文中的视频中找到了一些很棒的IBM代码：https://twitter.com/GantLaborde/status/1125735283343921152?s=20

我提取了该代码以制作一个ReactJS组件，用于检测此处的Rock / Paper / Scissors：https://github.com/GantMan/rps_tfjs_demo/blob/master/src/AdvancedModel.js

如果您想玩演示，请在此页面的底部：https://rps-tfjs.netlify.com/

所有这些都是开源的，似乎可以非常快速地实时检测单个对象。