processing/mask.pyi

import numpy as np
import torch
from PIL.Image import Image


def create_mask(image: Image, blur: int = ..., padding: int = ...) -> Image:
    """
        Create a mask from the input image.
        :param image: The input image.
        :param blur: Blur strength. Default is 0.
        :param padding: The padding around the object in the mask. Default is 0.
    """
    ...


def create_depth_map(image: np.ndarray, model: torch.nn.Module) -> Image:
    """
        Create a depth map from the input image.
    """
    ...