如何使用张量流对象检测API使用Mobilenet进行对象跟踪?

时间:2019-08-13 15:11:36

标签: opencv tensorflow object-detection-api video-tracking

我已经为自定义对象检测训练了对象检测API。现在,我想跟踪API检测到的对象。

1 个答案:

答案 0 :(得分:0)

看看SORT算法。它包含:

  • 用于预测新位置的卡尔曼滤波器
  • 与IOU距离匹配的匈牙利语跟踪和检测算法(或者您可以编写自己的条件)

在此之上,您可以攻击您想要的任何逻辑。该链接可能很有趣:

  • Deep SORT-对于每次检测,它会为图像区域建立描述符。该描述符与轨道描述符进行比较
  • SORT-具有FasterRCNN和SSD检测的简单SORT

在我们的项目中,我们使SORT适应于在检测之间使用模板匹配。如果性能对您至关重要,那么考虑这也是个好主意。