Different Augmentation on different Classes with .flow()

Hey there,

first of all i’m glad that i found that community and of cause the outstanding material around it. Thx for sharing it publicly!

Scenario:
First i’ll scatch a little scenario we can discuss about. Let’s say i’ve two classes i would like to classify. One class contains beautiful landscape pictures (class0), the other one ugly once (class1). In order to get more diversity for each of these classes i currently use a normal generator for image augmentation.

Here is my question:
Is it possible to define one generator which treats both classes differently?
I would like to have little augmentation for my class0 and kinda like go crazy on my class1.

I’m happy to read you thoughts about that!

If someone is interested in this topic i’ve opened a stack discussion about it.