wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-04-15 13:02:35 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

updated workflow

Browse Source
This commit is contained in:
Yousef Rafat 2026-04-10 13:49:29 +02:00
parent cf5faf8c17
commit 3aa8f900d6
1 changed files with 26 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
comfy_extras/nodes_mask.py
View File

@ -374,6 +374,29 @@ class GrowMask(IO.ComfyNode):
expand_mask = execute # TODO: remove expand_mask = execute # TODO: remove
class ClipVisionToMask(IO.ComfyNode):
@classmethod
def define_schema(cls):
return IO.Schema(
node_id="ClipVisionToMask",
inputs = [
IO.ClipVisionOutput.Input("clip_vision_output")
],
outputs = [IO.Mask.Output("mask")]
)
@classmethod
def execute(cls, clip_vision_output):
if not isinstance(clip_vision_output, torch.Tensor):
mask = clip_vision_output["last_hidden_state"]
mask = mask.sigmoid()
if mask.ndim == 3:
mask = mask.unsqueeze(0)
if mask.shape[1] != 1:
mask = mask.movedim(-1, 1)
return IO.NodeOutput(mask)
clip_vision_to_mask = execute
class ConcatMask(IO.ComfyNode): class ConcatMask(IO.ComfyNode):
@classmethod @classmethod
def define_schema(cls): def define_schema(cls):
@ -382,20 +405,13 @@ class ConcatMask(IO.ComfyNode):
search_aliases=["add mask", "concat mask", "merge mask"], search_aliases=["add mask", "concat mask", "merge mask"],
category="mask", category="mask",
inputs=[ inputs=[
IO.AnyType.Input("mask"), IO.Mask.Input("mask"),
IO.Image.Input("image"), IO.Image.Input("image"),
], ],
outputs=[IO.Image.Output("rgba_image"), IO.Mask.Output("input_mask")], outputs=[IO.Image.Output("rgba_image"), IO.Mask.Output("input_mask")],
) )
@classmethod @classmethod
def execute(cls, mask, image): def execute(cls, mask, image):
if not isinstance(mask, torch.Tensor):
mask = mask["last_hidden_state"]
mask = mask.sigmoid()
if mask.ndim == 3:
mask = mask.unsqueeze(0)
if mask.shape[1] != 1:
mask = mask.movedim(-1, 1)
if image.shape[-1] == 3: if image.shape[-1] == 3:
image = image.movedim(-1, 1) image = image.movedim(-1, 1)
target_h, target_w = image.shape[2], image.shape[3] target_h, target_w = image.shape[2], image.shape[3]
@ -471,7 +487,8 @@ class MaskExtension(ComfyExtension):
GrowMask, GrowMask,
ThresholdMask, ThresholdMask,
MaskPreview, MaskPreview,
ConcatMask ConcatMask,
ClipVisionToMask
] ]