Is it possible to apply data augmentation only on some classes ?

[Taechai]

I am currently working on a custom dataset of 20 classes and due to my unbalanced data I wanted to oversample some underdefined classes, so I thought it'd be better to make use of the default data augmentation provided by detectron. However I don't really know how to manipulate the main file or if it's even possible to only apply it on some classes.