tfm.vision.mask_ops.paste_instance_masks_v2

Paste instance masks to generate the image segmentation (v2).

tfm.vision.mask_ops.paste_instance_masks_v2(
    masks: np.ndarray,
    detected_boxes: np.ndarray,
    image_height: int,
    image_width: int
) -> np.ndarray

masks a numpy array of shape [N, mask_height, mask_width] representing the instance masks w.r.t. the detected_boxes.
detected_boxes a numpy array of shape [N, 4] representing the reference bounding boxes.
image_height an integer representing the height of the image.
image_width an integer representing the width of the image.

segms a numpy array of shape [N, image_height, image_width] representing the instance masks pasted on the image canvas.