age], or List[np.ndarray]):
Image, NumPy array or tensor representing an image batch to mask image. White pixels in the mask
are repainted while black pixels are preserved. If mask_image is a PIL image, it is converted to a
single channel (luminance) before use. If it’s a NumPy array or PyTorch tensor, it should contain one
color channel (L) instead of 3, so the expected shape for PyTorch tensor would be (B, 1, H, W), (B, H, W), (1, H, W), (H, 