search

Unity-Technologies / pysolotools

Python toolchain for SOLO.
https://Unity-Technologies.github.io/pysolotools
Other
39 stars 14 forks source link

Adding support for keypoint names/labels other than the ones used in the COCO dataset #139

Open hoel-bagard opened 1 year ago

hoel-bagard commented 1 year ago

Hi, This is the PR corresponding to the issue with the same name here.

This PR aims to remove the hard-coded COCO category names, and to replace them by the ones present in the dataset.\ This allows to use the tool for datasets where the keypoints are not defined for humanoid shapes.

The PR also adds the keypoints names and skeleton to the first category of the COCO json file. This does not guarantee that they are attached to the correct category, but this will be fine if only one type of object is present (and it is likely better than not having the information at all).

unity-cla-assistant commented 1 year ago

CLA assistant check
All committers have signed the CLA.