search

albumentations-team / albumentations

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125
https://albumentations.ai
MIT License
13.8k stars 1.61k forks source link

keypoints augmentation for invisible keypoints #1092

Open puruBHU opened 2 years ago

puruBHU commented 2 years ago

I am working on hand pose estimation and planning to use albumentations library for augmentations. On testing albumentation library for a few samples from a dataset that it has a problem augmenting invisible keypoints which are represented as either -1 or 0.

The augmentation, in this case, is either unsupported or is wrong. I need help to use augmentations for keypoints in this case.

Dipet commented 2 years ago

Have you tried setting remove_invisible = False? A.KeypointParams("xy", remove_invisible=False)