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

时间:2018-11-13 14:42:46

标签: tensorflow object-detection-api

我在Windows系统上使用tensorflow object detection api,并且运行正常。我现在使用的是ssd_mobilenet_v1_coco_11_06_2017的预训练模型。它可以轻松检测给定输入视频中的所有对象。我想记录每个对象的时间,所以可以说,如果在视频中有一辆汽车从00:05进入并离开00:15,这意味着它已经进入10secs

要实现这一点,我一直在寻找API返回的每个检测到的对象的ID之类的东西,以便我可以从代码中启动计时器以计算对象的时间。 API中是否已经为此建立了任何功能??

2 个答案:

答案 0 :(得分:0)

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

答案 1 :(得分:0)

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

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

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

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