Tensorflow：对象检测API是否返回检测到的对象ID

Question

我在 Windows 系统上使用tensorflowobject detection api，它一直运行良好。截至目前，我正在使用预训练模型ssd_mobilenet_v1_coco_11_06_2017。它很容易检测给定输入视频中的所有对象。我想记录每个对象的时间，所以可以说，如果在视频中汽车在00:05 进入并在00:15 离开，这意味着它在10secs。

为了实现这一点，我正在查看 API 返回的每个检测到的对象的 id 之类的东西，以便我可以从代码中启动一个计时器来计算对象的时间。 API 中是否已经为此构建了任何功能？

Answer 1

Tensorflow 对象检测不提供此类功能，但您可以使用 KFC 算法（使用 Open CV 很容易获得）来跟踪对象。

https://www.docs.opencv.org/3.4.1/d2/dff/classcv_1_1TrackerKCF.html

或者您可以在使用卡尔曼滤波器但易于集成的对象检测 API 之上实现 SORT。

https://github.com/abewley/sort/blob/master/sort.py

Answer 2

Tensorflow 对象检测 API 目前不跟踪帧之间的对象。