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

Update curve_nodes.py

Browse Source
This commit is contained in:
kijai 2024-05-12 02:08:53 +03:00
parent 9991155130
commit 5631cd0146
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
nodes/curve_nodes.py
View File

@ -305,8 +305,8 @@ Locations are center locations.
class CreateShapeImageOnPath:
RETURN_TYPES = ("IMAGE", )
RETURN_NAMES = ("image", )
RETURN_TYPES = ("IMAGE", "MASK",)
RETURN_NAMES = ("image","mask", )
FUNCTION = "createshapemask"
CATEGORY = "KJNodes/image"
DESCRIPTION = """
@ -348,7 +348,8 @@ Locations are center locations.
coordinates = json.loads(coordinates)
batch_size = len(coordinates)
out = []
images_list = []
masks_list = []
if len(size_multiplier) != batch_size:
size_multiplier = size_multiplier * (batch_size // len(size_multiplier)) + size_multiplier[:batch_size % len(size_multiplier)]
@ -386,9 +387,12 @@ Locations are center locations.
image = pil2tensor(image)
image = image * intensity
out.append(image)
outstack = torch.cat(out, dim=0)
return (outstack,)
mask = image[:, :, :, 0]
masks_list.append(mask)
images_list.append(image)
out_images = torch.cat(images_list, dim=0).cpu().float()
out_masks = torch.cat(masks_list, dim=0)
return (out_images, out_masks)
class CreateTextOnPath: