How to make different data augmentation according to each class

Hello everybody,

I can’t find a way to perform adjusted data augmentation according to each class.

Let’s assume I would like to do 3 different kind of data augmentation (aug1,aug2 and aug3)
Let’s assume also that I have 2 different classes (class1 and class2).

I would like to be able to do for example:
aug1, aug2 and aug3 for class1
aug1 and aug3 for class2

Does anyone see a way to do it ?

Thanks for your answer and best regards

1 Like