ls_mlkit.util.mask.image_masker module

class ls_mlkit.util.mask.image_masker.ImageMasker(ndim_mini_micro_shape: int = 0, **kwargs: dict[Any, Any])

Bases: Masker

check_mask_shape(x: Tensor, mask: Tensor)

check whether the shape of mask is as expected