xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-16 00:04:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update curve_nodes.py

Browse Source
This commit is contained in:
kijai 2025-01-27 12:14:45 +02:00
parent 3f141b8f1c
commit 2abf557e3d
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
nodes/curve_nodes.py
View File

@ -1452,15 +1452,14 @@ you can clear the image from the context menu by right clicking on the canvas
"ui": {"bg_image": [img_base64]}, "ui": {"bg_image": [img_base64]},
"result": (json.dumps(pos_coordinates), json.dumps(neg_coordinates), bboxes, mask_tensor, cropped_image) "result": (json.dumps(pos_coordinates), json.dumps(neg_coordinates), bboxes, mask_tensor, cropped_image)
} }
import cv2
class CutAndDragOnPath: class CutAndDragOnPath:
RETURN_TYPES = ("IMAGE", "MASK",) RETURN_TYPES = ("IMAGE", "MASK",)
RETURN_NAMES = ("image","mask", ) RETURN_NAMES = ("image","mask", )
FUNCTION = "createshapemask" FUNCTION = "cutanddrag"
CATEGORY = "KJNodes/image" CATEGORY = "KJNodes/image"
DESCRIPTION = """ DESCRIPTION = """
Cuts the masked area from the image, and drags it along the path. Cuts the masked area from the image, and drags it along the path. If inpaint is enabled, and no bg_image is provided, the cut area is filled using cv2 TELEA algorithm.
""" """
@classmethod @classmethod
@ -1479,7 +1478,7 @@ Cuts the masked area from the image, and drags it along the path.
} }
} }
def createshapemask(self, image, coordinates, mask, frame_width, frame_height, inpaint, bg_image=None): def cutanddrag(self, image, coordinates, mask, frame_width, frame_height, inpaint, bg_image=None):
# Parse coordinates # Parse coordinates
if len(coordinates) < 10: if len(coordinates) < 10:
coords_list = [] coords_list = []
@ -1518,6 +1517,7 @@ Cuts the masked area from the image, and drags it along the path.
background = input_image.copy() background = input_image.copy()
# Inpaint the cut area # Inpaint the cut area
if inpaint: if inpaint:
import cv2
border = 5 # Create small border around cut area for better inpainting border = 5 # Create small border around cut area for better inpainting
fill_mask = Image.new("L", background.size, 0) fill_mask = Image.new("L", background.size, 0)
draw = ImageDraw.Draw(fill_mask) draw = ImageDraw.Draw(fill_mask)