openvinotoolkit / datumaro

Dataset Management Framework, a Python library and a CLI tool to build, analyze and manage Computer Vision datasets.
https://openvinotoolkit.github.io/datumaro/
MIT License
542 stars 135 forks source link

Add support for Ultralytics Yolo tracking format #1666

Open gboeer opened 1 week ago

gboeer commented 1 week ago

With newer yolo models by Ultralytics it is possible to perform object tracking. Currently, it is not possible to handle such tracking outputs with datumaro.

In particular, the output of Yolo can contain a tracking id for each bounding box like: <class_id> <x_center> <y_center> <width> <height> <track_id>

It would be good to be able to load such datasets and especially to convert them to other dataset formats which allow having a tracking id as well.

jihyeonyi commented 1 week ago

Thanks for your interest in datumaro project. We will consider supporting the new format.