Albumentations Compose using OneOf on Image and Mask

Viewed 12

i set up my transformation like this:

aug_albu = A.Compose([A.OneOf([A.HorizontalFlip(p=1),
                              A.RandomRotate90(p=1),
                              A.VerticalFlip(p=1)])])  

and then calling it like:

augmented = aug_albu(image=img, mask=depth)
img = augmented["image"]
depth= augmented["mask"] 

So i wonder if it apllies the same transformation to the image and the mask from the given transforms in the OneOf-Part.

0 Answers
Related