Tensorflow: API обнаружения объекта возвращает идентификатор обнаруженного объекта

Я использую tensorflowobject detection api в системе Windows, и она работала нормально. Я использую предварительно обученную модель, которая сейчас ssd_mobilenet_v1_coco_11_06_2017, Это легко обнаруживает все объекты в данном входном видео. Я хотел записать время каждого объекта, так скажем, если в видео автомобиль входит в 00:05 и уходит в 00:15 а это значит, что это было в течение 10secs,

Чтобы добиться этого, я искал, есть ли что-нибудь вроде идентификатора каждого обнаруженного объекта, который возвращает API, чтобы я мог запустить таймер из кода для расчета времени объекта. Есть ли уже встроенная функциональность для этого в API.?

2 ответа

Решение

Обнаружение Tensorflow Object не предоставляет такой функциональности, но вы можете использовать алгоритм KFC (легко доступный с помощью Open CV) для отслеживания объекта.

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

или Вы можете реализовать SORT выше API обнаружения объектов, который использует Kalman Filter, но легко интегрируется.

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

API обнаружения объектов Tensorflow в настоящее время не отслеживает объекты между кадрами.

Другие вопросы по тегам