Augment only parts of the image

[Testbild]

Hi everyone,

I have the following issue: I want to augment only part of the image. Essentially I want to do what is shown here in this example: https://imgaug.readthedocs.io/en/latest/source/examples_segmentation_maps.html#a-simple-example only that I want to augment the image only where the segmentation mask is. And frankly speaking I am quite stuck on how to do this. I get the results from the example, but I cannot manage to make a partial augmentation on the image.

I hope someone can help me here, thank you :)