Spaces:

gunnit
/

damostudio

Runtime error

App Files Files Community

damostudio / processing /mask.pyi

mohammadhakimi

add blur strength in inputs

8c7592a over 2 years ago

Raw History Blame Contribute Delete

526 Bytes

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.
    """
    ...

	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.
	"""
	...