mixmo.augmentations.mixing_blocks._cow_mask

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

Compute masks for CowMix lam is overridden by Cowmask’s parameters https://github.com/google-research/google-research/tree/master/milking_cowmask/masking