mixmo.augmentations.mixing_blocks._stack_mask

mixmo.augmentations.mixing_blocks._stack_mask(input_size, lam, config_mix)[source]

Compute masks for Channel/Horizontal/Vertical concat (number of images) x channel x (image width) x (image height)