mixmo.augmentations.mixing_blocks._cutmix_mask

mixmo.augmentations.mixing_blocks._cutmix_mask(input_size, lam, config_mix=None)[source]

Compute masks for CutMix