SeamlessClone (simple)

Documentation

• Class name: SeamlessClone (simple)
• Category: Bmad/CV/C.Photography
• Output node: False

This node provides a simplified interface for performing seamless cloning operations on images, automatically determining the center of the source mask for the cloning process. It leverages the capabilities of a more complex seamless cloning node to offer an easier-to-use option for integrating one image into another with visually coherent results.

Input types

Required

• dst
  • The destination image onto which the source image will be cloned.
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• src
  • The source image to be cloned onto the destination image.
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• src_mask
  • A mask image indicating the region of the source image to be cloned.
  • Comfy dtype: IMAGE
  • Python dtype: torch.Tensor
• flag
  • Specifies the cloning mode to be used, which can affect the blending behavior.
  • Comfy dtype: COMBO[STRING]
  • Python dtype: str

Output types

• image
  • Comfy dtype: IMAGE
  • The result of the seamless cloning operation, with the source image integrated into the destination image.
  • Python dtype: torch.Tensor

Usage tips

• Infra type: GPU
• Common nodes: unknown

Source code

class SeamlessCloneSimpler:
    @classmethod
    def INPUT_TYPES(cls):
        return {
            "required": {
                "dst": ("IMAGE",),
                "src": ("IMAGE",),
                "src_mask": ("IMAGE",),
                "flag": (SeamlessClone.clone_modes, {"default": SeamlessClone.clone_modes[0]}),
            },
        }

    RETURN_TYPES = ("IMAGE",)
    FUNCTION = "paste"
    CATEGORY = f"{cv_category_path}/C.Photography"

    @staticmethod
    def get_center(cv_mask):
        br = cv.boundingRect(cv_mask)
        return br[0] + br[2] // 2, br[1] + br[3] // 2

    def paste(self, src, dst, src_mask, flag):
        src_mask_cv = tensor2opencv(src_mask, 1)
        cx, cy = SeamlessCloneSimpler.get_center(src_mask_cv)
        sc = SeamlessClone()
        return sc.paste(src, dst, src_mask, flag, cx, cy)