orch.FloatTensor], List[PIL.Image.Image], 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,