def METHOD_NAME(
        self,
        masks: paddle.Tensor,
        input_size: Tuple[int, ...],
        original_size: Tuple[int, ...], ) -> paddle.Tensor:
    """
    Remove padding and upscale masks to the original image size.
    Arguments:
      masks (paddle.Tensor): Batched masks from the mask_decoder,
        in BxCxHxW format.
      input_size (tuple(int, int)): The size of the image input to the
        model, in (H, W) format. Used to remove padding.
      original_size (tuple(int, int)): The original size of the image
        before resizing for input to the model, in (H, W) format.
    Returns:
      (paddle.Tensor): Batched masks in BxCxHxW format, where (H, W)
        is given by original_size.
    """
    masks = F.interpolate(
        masks,
        (self.image_encoder.img_size, self.image_encoder.img_size),
        mode="bilinear",
        align_corners=False, )
    masks = masks[..., :input_size[0], :input_size[1]]
    masks = F.interpolate(
        masks, original_size, mode="bilinear", align_corners=False)
    return masks