29 lines
897 B
Python
29 lines
897 B
Python
from .utils.apply_blur import apply_blur
|
|
from .utils.apply_padding import apply_padding
|
|
from .utils.tensor_to_image import tensor_to_image
|
|
from .utils.image_to_tensor import image_to_tensor
|
|
from .utils.image_to_mask import image_to_mask
|
|
from .utils.image_composite import image_composite
|
|
|
|
class ImageComposite:
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"image": ("IMAGE",),
|
|
"mask": ("MASK",),
|
|
"color": (["black", "white","transparent"],)
|
|
}
|
|
}
|
|
CATEGORY = "TrollSuite/image"
|
|
RETURN_TYPES = ("IMAGE","MASK",)
|
|
RETURN_NAMES = ("image","mask",)
|
|
FUNCTION = "ImageComposite"
|
|
|
|
|
|
def ImageComposite(self, image, mask, color):
|
|
image = image_composite(tensor_to_image(image), tensor_to_image(mask), color)
|
|
image = image_to_tensor(image)
|
|
return (image, mask,)
|