wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-15 03:27:24 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

Move mask category under image

Browse Source
This commit is contained in:
Alexis Rolland 2026-05-10 15:13:39 +08:00
parent 5b8054ea0f
commit d28d760825
3 changed files with 17 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
comfy_extras/nodes_mask.py
View File

@ -127,7 +127,7 @@ class MaskToImage(IO.ComfyNode):
node_id="MaskToImage", node_id="MaskToImage",
search_aliases=["convert mask"], search_aliases=["convert mask"],
display_name="Convert Mask to Image", display_name="Convert Mask to Image",
category="mask", category="image/mask",
inputs=[ inputs=[
IO.Mask.Input("mask"), IO.Mask.Input("mask"),
], ],
@ -149,7 +149,7 @@ class ImageToMask(IO.ComfyNode):
node_id="ImageToMask", node_id="ImageToMask",
search_aliases=["extract channel", "channel to mask"], search_aliases=["extract channel", "channel to mask"],
display_name="Convert Image to Mask", display_name="Convert Image to Mask",
category="mask", category="image/mask",
inputs=[ inputs=[
IO.Image.Input("image"), IO.Image.Input("image"),
IO.Combo.Input("channel", options=["red", "green", "blue", "alpha"]), IO.Combo.Input("channel", options=["red", "green", "blue", "alpha"]),
@ -172,7 +172,8 @@ class ImageColorToMask(IO.ComfyNode):
return IO.Schema( return IO.Schema(
node_id="ImageColorToMask", node_id="ImageColorToMask",
search_aliases=["color keying", "chroma key"], search_aliases=["color keying", "chroma key"],
category="mask", display_name="Convert Image Color to Mask",
category="image/mask",
inputs=[ inputs=[
IO.Image.Input("image"), IO.Image.Input("image"),
IO.Int.Input("color", default=0, min=0, max=0xFFFFFF, step=1, display_mode=IO.NumberDisplay.number), IO.Int.Input("color", default=0, min=0, max=0xFFFFFF, step=1, display_mode=IO.NumberDisplay.number),
@ -194,8 +195,9 @@ class SolidMask(IO.ComfyNode):
@classmethod @classmethod
def define_schema(cls): def define_schema(cls):
return IO.Schema( return IO.Schema(
node_id="SolidMask", node_id="CreateSolidMask",
category="mask", display_name="Create Solid Mask",
category="image/mask",
inputs=[ inputs=[
IO.Float.Input("value", default=1.0, min=0.0, max=1.0, step=0.01), IO.Float.Input("value", default=1.0, min=0.0, max=1.0, step=0.01),
IO.Int.Input("width", default=512, min=1, max=nodes.MAX_RESOLUTION, step=1), IO.Int.Input("width", default=512, min=1, max=nodes.MAX_RESOLUTION, step=1),
@ -219,7 +221,7 @@ class InvertMask(IO.ComfyNode):
node_id="InvertMask", node_id="InvertMask",
search_aliases=["reverse mask", "flip mask"], search_aliases=["reverse mask", "flip mask"],
display_name="Invert Mask", display_name="Invert Mask",
category="mask", category="image/mask",
inputs=[ inputs=[
IO.Mask.Input("mask"), IO.Mask.Input("mask"),
], ],
@ -241,7 +243,7 @@ class CropMask(IO.ComfyNode):
node_id="CropMask", node_id="CropMask",
search_aliases=["cut mask", "extract mask region", "mask slice"], search_aliases=["cut mask", "extract mask region", "mask slice"],
display_name="Crop Mask", display_name="Crop Mask",
category="mask", category="image/mask",
inputs=[ inputs=[
IO.Mask.Input("mask"), IO.Mask.Input("mask"),
IO.Int.Input("x", default=0, min=0, max=nodes.MAX_RESOLUTION, step=1), IO.Int.Input("x", default=0, min=0, max=nodes.MAX_RESOLUTION, step=1),
@ -268,7 +270,7 @@ class MaskComposite(IO.ComfyNode):
node_id="MaskComposite", node_id="MaskComposite",
search_aliases=["combine masks", "blend masks", "layer masks", "masks composition"], search_aliases=["combine masks", "blend masks", "layer masks", "masks composition"],
display_name="Combine Masks", display_name="Combine Masks",
category="mask", category="image/mask",
inputs=[ inputs=[
IO.Mask.Input("destination"), IO.Mask.Input("destination"),
IO.Mask.Input("source"), IO.Mask.Input("source"),
@ -319,7 +321,7 @@ class FeatherMask(IO.ComfyNode):
node_id="FeatherMask", node_id="FeatherMask",
search_aliases=["soft edge mask", "blur mask edges", "gradient mask edge"], search_aliases=["soft edge mask", "blur mask edges", "gradient mask edge"],
display_name="Feather Mask", display_name="Feather Mask",
category="mask", category="image/mask",
inputs=[ inputs=[
IO.Mask.Input("mask"), IO.Mask.Input("mask"),
IO.Int.Input("left", default=0, min=0, max=nodes.MAX_RESOLUTION, step=1), IO.Int.Input("left", default=0, min=0, max=nodes.MAX_RESOLUTION, step=1),
@ -367,7 +369,7 @@ class GrowMask(IO.ComfyNode):
node_id="GrowMask", node_id="GrowMask",
search_aliases=["expand mask", "shrink mask"], search_aliases=["expand mask", "shrink mask"],
display_name="Grow Mask", display_name="Grow Mask",
category="mask", category="image/mask",
inputs=[ inputs=[
IO.Mask.Input("mask"), IO.Mask.Input("mask"),
IO.Int.Input("expand", default=0, min=-nodes.MAX_RESOLUTION, max=nodes.MAX_RESOLUTION, step=1), IO.Int.Input("expand", default=0, min=-nodes.MAX_RESOLUTION, max=nodes.MAX_RESOLUTION, step=1),
@ -403,7 +405,8 @@ class ThresholdMask(IO.ComfyNode):
return IO.Schema( return IO.Schema(
node_id="ThresholdMask", node_id="ThresholdMask",
search_aliases=["binary mask"], search_aliases=["binary mask"],
category="mask", display_name="Threshold Mask",
category="image/mask",
inputs=[ inputs=[
IO.Mask.Input("mask"), IO.Mask.Input("mask"),
IO.Float.Input("value", default=0.5, min=0.0, max=1.0, step=0.01), IO.Float.Input("value", default=0.5, min=0.0, max=1.0, step=0.01),
@ -429,7 +432,7 @@ class MaskPreview(IO.ComfyNode):
node_id="MaskPreview", node_id="MaskPreview",
search_aliases=["show mask", "view mask", "inspect mask", "debug mask"], search_aliases=["show mask", "view mask", "inspect mask", "debug mask"],
display_name="Preview Mask", display_name="Preview Mask",
category="mask", category="image/mask",
description="Saves the input images to your ComfyUI output directory.", description="Saves the input images to your ComfyUI output directory.",
inputs=[ inputs=[
IO.Mask.Input("mask"), IO.Mask.Input("mask"),

2
comfy_extras/nodes_post_processing.py
View File

@ -595,7 +595,7 @@ class BatchMasksNode(io.ComfyNode):
node_id="BatchMasksNode", node_id="BatchMasksNode",
search_aliases=["combine masks", "stack masks", "merge masks"], search_aliases=["combine masks", "stack masks", "merge masks"],
display_name="Batch Masks", display_name="Batch Masks",
category="mask", category="image/mask",
inputs=[ inputs=[
io.Autogrow.Input("masks", template=autogrow_template) io.Autogrow.Input("masks", template=autogrow_template)
], ],

2
nodes.py
View File

@ -1771,7 +1771,7 @@ class LoadImageMask(LoadImage):
} }
} }
CATEGORY = "mask" CATEGORY = "image/mask"
RETURN_TYPES = ("MASK",) RETURN_TYPES = ("MASK",)
FUNCTION = "load_image_mask" FUNCTION = "load_image_mask"