ImageDataAugmenter does the augmentation during image loading, which means it doesn’t rebalance your data. It applies the selected augmentation on the fly.
If you would like to rebalance your data, you need to apply the augmentation offline and adjust the number of augmented samples accordingly. You can design your own custom function to do that. I refer you to our WB augmenter here .
Another option is to use a weighted loss func that penalizes more for mistakes in the classes that have lower training examples.
