diff --git a/comfy_extras/nodes_mask.py b/comfy_extras/nodes_mask.py index c9b2a84d9..50c763d43 100644 --- a/comfy_extras/nodes_mask.py +++ b/comfy_extras/nodes_mask.py @@ -127,7 +127,7 @@ class MaskToImage(IO.ComfyNode): node_id="MaskToImage", search_aliases=["convert mask"], display_name="Convert Mask to Image", - category="mask", + category="image/mask", inputs=[ IO.Mask.Input("mask"), ], @@ -149,7 +149,7 @@ class ImageToMask(IO.ComfyNode): node_id="ImageToMask", search_aliases=["extract channel", "channel to mask"], display_name="Convert Image to Mask", - category="mask", + category="image/mask", inputs=[ IO.Image.Input("image"), IO.Combo.Input("channel", options=["red", "green", "blue", "alpha"]), @@ -172,7 +172,8 @@ class ImageColorToMask(IO.ComfyNode): return IO.Schema( node_id="ImageColorToMask", search_aliases=["color keying", "chroma key"], - category="mask", + display_name="Convert Image Color to Mask", + category="image/mask", inputs=[ IO.Image.Input("image"), 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 def define_schema(cls): return IO.Schema( - node_id="SolidMask", - category="mask", + node_id="CreateSolidMask", + display_name="Create Solid Mask", + category="image/mask", inputs=[ 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), @@ -219,7 +221,7 @@ class InvertMask(IO.ComfyNode): node_id="InvertMask", search_aliases=["reverse mask", "flip mask"], display_name="Invert Mask", - category="mask", + category="image/mask", inputs=[ IO.Mask.Input("mask"), ], @@ -241,7 +243,7 @@ class CropMask(IO.ComfyNode): node_id="CropMask", search_aliases=["cut mask", "extract mask region", "mask slice"], display_name="Crop Mask", - category="mask", + category="image/mask", inputs=[ IO.Mask.Input("mask"), 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", search_aliases=["combine masks", "blend masks", "layer masks", "masks composition"], display_name="Combine Masks", - category="mask", + category="image/mask", inputs=[ IO.Mask.Input("destination"), IO.Mask.Input("source"), @@ -319,7 +321,7 @@ class FeatherMask(IO.ComfyNode): node_id="FeatherMask", search_aliases=["soft edge mask", "blur mask edges", "gradient mask edge"], display_name="Feather Mask", - category="mask", + category="image/mask", inputs=[ IO.Mask.Input("mask"), 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", search_aliases=["expand mask", "shrink mask"], display_name="Grow Mask", - category="mask", + category="image/mask", inputs=[ IO.Mask.Input("mask"), 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( node_id="ThresholdMask", search_aliases=["binary mask"], - category="mask", + display_name="Threshold Mask", + category="image/mask", inputs=[ IO.Mask.Input("mask"), 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", search_aliases=["show mask", "view mask", "inspect mask", "debug mask"], display_name="Preview Mask", - category="mask", + category="image/mask", description="Saves the input images to your ComfyUI output directory.", inputs=[ IO.Mask.Input("mask"), diff --git a/comfy_extras/nodes_post_processing.py b/comfy_extras/nodes_post_processing.py index be6ec7434..c30052d76 100644 --- a/comfy_extras/nodes_post_processing.py +++ b/comfy_extras/nodes_post_processing.py @@ -595,7 +595,7 @@ class BatchMasksNode(io.ComfyNode): node_id="BatchMasksNode", search_aliases=["combine masks", "stack masks", "merge masks"], display_name="Batch Masks", - category="mask", + category="image/mask", inputs=[ io.Autogrow.Input("masks", template=autogrow_template) ], diff --git a/nodes.py b/nodes.py index 5755f0bb8..cbca01c03 100644 --- a/nodes.py +++ b/nodes.py @@ -1771,7 +1771,7 @@ class LoadImageMask(LoadImage): } } - CATEGORY = "mask" + CATEGORY = "image/mask" RETURN_TYPES = ("MASK",) FUNCTION = "load_image_mask"