search

VlSomers / keypoint_promptable_reidentification

[ECCV24] Keypoint Promptable Re-Identification: SOTA ReID method robust to occlusions and multi-person ambiguity
Other
87 stars 5 forks source link

Code for generating keypoint annotations (.json) #3

Open Sunnie0101 opened 2 months ago

Sunnie0101 commented 2 months ago

Hi, @VlSomers, is there any instruction or code for generating keypoint annotations (.json). I want to try this on my own dataset, thanks!

VlSomers commented 2 months ago

Hi @Sunnie0101, what is your dataset format? Is it a reid dataset or a tracking dataset?

Sunnie0101 commented 2 months ago

It is a reid dataset. Is it available for both reid and tracking datasets?

VlSomers commented 2 months ago

To generate the keypoints for existing ReID datasets, I used OpenPifPaf. You can have a look at this BPBreID pull request for some code to generate the human parsing labels using pifpaf and maskRCNN. You would need to modify this code to also save the keypoints as json files. If you want to be a contributor to this repository, please feel free to open a pull request with you keypoint generation code! ;) Let me know if you have other questions

Sunnie0101 commented 2 months ago

thanks! I successfully generated the keypoints. And I'm wondering that are the negative keypoints created by Openpifpaf or manual labeling?

VlSomers commented 2 months ago

Negative keypoints are others skeletons in the same image crop, i.e. skeletons from other persons (e.g. occluders) that the ReID target. You can have a deeper look at the demo to see how they are defined.