this means we can add any type of augmentation in the function above and during these two lines:
image_heavy = augmented[‘image’]
mask_heavy = augmented[‘mask’]
some operations will apply to images and others to mask based on the type of augmentation.
is this right?