Tensorflow:对象检测 API returns 是否检测到对象 id

Tensorflow: Does object detection API returns detected object id

我在 windows 系统上使用 tensorflow object detection api,它运行良好。我现在使用的是 ssd_mobilenet_v1_coco_11_06_2017 的预训练模型。它很容易检测给定输入视频中的所有对象。我想记录每个物体的时间,所以假设,如果在视频中汽车在 00:05 进入并在 00:15 离开,这意味着它在 10secs

为了实现这一点,我正在寻找 API returns 每个检测到的对象的 ID 之类的东西,这样我就可以从代码中启动一个计时器来计算一个时间目的。 API.?

中是否已经为此构建了任何功能?

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

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

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

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

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