Skip to content

A set of nodes for ComfyUI that can composite layer and mask to achieve Photoshop like functionality.

License

Notifications You must be signed in to change notification settings

chflame163/ComfyUI_LayerStyle

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

ComfyUI Layer Style

中文说明点这里

商务合作请联系email chflame@163.com.

For business cooperation, please contact email chflame@163.com.

A set of nodes for ComfyUI that can composite layer and mask to achieve Photoshop like functionality.

It migrate some basic functions of PhotoShop to ComfyUI, aiming to centralize the workflow and reduce the frequency of software switching.

image
*this workflow (title_example_workflow.json) is in the workflow directory.

Example workflow

Some JSON workflow files in the workflow directory, That's examples of how these nodes can be used in ComfyUI.

How to install

(Taking ComfyUI official portable package and Aki ComfyUI package as examples, please modify the dependency environment directory for other ComfyUI environments)

Install plugin

  • Recommended use ComfyUI Manager for installation.

  • Or open the cmd window in the plugin directory of ComfyUI, like ComfyUI\custom_nodes,type

git clone https://github.com/chflame163/ComfyUI_LayerStyle.git
  • Or download the zip file and extracted, copy the resulting folder to ComfyUI\custom_ Nodes

Install dependency packages

  • for ComfyUI official portable package, double-click the install_requirements.bat in the plugin directory, for Aki ComfyUI package double-click on the install_requirements_aki.bat in the plugin directory, and wait for the installation to complete.

  • Or install dependency packages, open the cmd window in the ComfyUI_LayerStyle plugin directory like ComfyUI\custom_ Nodes\ComfyUI_LayerStyle and enter the following command,

  for ComfyUI official portable package, type:

..\..\..\python_embeded\python.exe -s -m pip install .\whl\docopt-0.6.2-py2.py3-none-any.whl
..\..\..\python_embeded\python.exe -s -m pip install .\whl\hydra_core-1.3.2-py3-none-any.whl
..\..\..\python_embeded\python.exe -s -m pip install -r requirements.txt
.\repair_dependency.bat

  for Aki ComfyUI package, type:

..\..\python\python.exe -s -m pip install .\whl\docopt-0.6.2-py2.py3-none-any.whl
..\..\python\python.exe -s -m pip install .\whl\hydra_core-1.3.2-py3-none-any.whl
..\..\python\python.exe -s -m pip install -r requirements.txt
.\repair_dependency.bat
  • Restart ComfyUI.

Download Model Files

From BaiduNetdisk download all files and copy them to ComfyUI\models folder. This link provides all the model files required for this plugin. Or download the model file according to the instructions of each node.

Common Issues

If the node cannot load properly or there are errors during use, please check the error message in the ComfyUI terminal window. The following are common errors and their solutions.

Warning: xxxx.ini not found, use default xxxx..

This warning message indicates that the ini file cannot be found and does not affect usage. If you do not want to see these warnings, please modify all *.ini.example files in the plugin directory to *.ini.

ModuleNotFoundError: No module named 'psd_tools'

This error is that the psd_tools were not installed correctly.

Solution:

  • Close ComfyUI and open the terminal window in the plugin directory and execute the following command: ../../../python_embeded/python.exe -s -m pip install psd_tools If error occurs during the installation of psd_tool, such as ModuleNotFoundError: No module named 'docopt' , please download docopt's whl and manual install it. execute the following command in terminal window: ../../../python_embeded/python.exe -s -m pip install path/docopt-0.6.2-py2.py3-none-any.whl the path is path name of whl file.

Cannot import name 'guidedFilter' from 'cv2.ximgproc'

This error is caused by incorrect version of the opencv-contrib-python package,or this package is overwriteen by other opencv packages.

NameError: name 'guidedFilter' is not defined

The reason for the problem is the same as above.

Cannot import name 'VitMatteImageProcessor' from 'transformers'

This error is caused by the low version of transformers package.

insightface Loading very slow

This error is caused by the low version of protobuf package.

For the issues with the above three dependency packages, please double click repair_dependency.bat (for Official ComfyUI Protable) or repair_dependency_aki.bat (for ComfyUI-aki-v1.x) in the plugin folder to automatically fix them.

onnxruntime::python::CreateExecutionProviderInstance CUDA_PATH is set but CUDA wasn't able to be loaded. Please install the correct version of CUDA and cuDNN as mentioned in the GPU requirements page

Solution: Reinstall the onnxruntime dependency package.

Error loading model xxx: We couldn't connect to huggingface.co ...

Check the network environment. If you cannot access huggingface.co normally in China, try modifying the huggingface_hub package to force the use hf_mirror.

  • Find constants.py in the directory of huggingface_hub package (usually Lib/site packages/huggingface_hub in the virtual environment path), Add a line after import os
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'

ValueError: Trimap did not contain foreground values (xxxx...)

This error is caused by the mask area being too large or too small when using the PyMatting method to handle the mask edges.

Solution:

  • Please adjust the parameters to change the effective area of the mask. Or use other methods to handle the edges.

Requests.exceptions.ProxyError: HTTPSConnectionPool(xxxx...)

When this error has occurred, please check the network environment.

Update

**If the dependency package error after updating, please double clicking repair_dependency.bat (for Official ComfyUI Protable) or repair_dependency_aki.bat (for ComfyUI-aki-v1.x) in the plugin folder to reinstall the dependency packages.

Description

Nodes are divided into 5 groups according to their functions: LayerStyle, LayerColor, LayerMask, LayerUtility and LayerFilter.

  • LayerStyle nodes provides layer styles that mimic Adobe Photoshop. image
  • LayerColor node group provides color adjustment functionality. image
  • LayerMask nodes provides mask assistance tools. image
  • LayerUtility nodes provides auxiliary nodes related to layer composit tools and workflows. image
  • LayerFilter nodes provides image effect filters. image

image
image

Generate shadow image

Node options: image

  • background_image1: The background image.
  • layer_image1: Layer image for composite.
  • layer_mask1,2: Mask for layer_image, shadows are generated according to their shape.
  • invert_mask: Whether to reverse the mask.
  • blend_mode3: Blending mode of shadows.
  • opacity: Opacity of shadow.
  • distance_x: Horizontal offset of shadow.
  • distance_y: Vertical offset of shadow.
  • grow: Shadow expansion amplitude.
  • blur: Shadow blur level.
  • shadow_color4: Shadow color.
  • note

Generate outer glow image

Node options: image

  • background_image1: The background image.
  • layer_image1: Layer image for composite.
  • layer_mask1,2: Mask for layer_image, grow are generated according to their shape.
  • invert_mask: Whether to reverse the mask.
  • blend_mode3: Blending mode of glow.
  • opacity: Opacity of glow.
  • brightness: Luminance of light.
  • glow_range: range of glow.
  • blur:blur of glow.
  • light_color4: Center part color of glow.
  • glow_color4: Edge part color of glow.
  • note

Generate inner shadow image

Node options: image

  • background_image1: The background image.
  • layer_image1: Layer image for composite.
  • layer_mask1,2: Mask for layer_image, shadows are generated according to their shape.
  • invert_mask: Whether to reverse the mask.
  • blend_mode3: Blending mode of shadows.
  • opacity: Opacity of shadow.
  • distance_x: Horizontal offset of shadow.
  • distance_y: Vertical offset of shadow.
  • grow: Shadow expansion amplitude.
  • blur: Shadow blur level.
  • shadow_color4: Shadow color.
  • note

Generate inner glow image

Node options:
image

  • background_image1: The background image.
  • layer_image1: Layer image for composite.
  • layer_mask1,2: Mask for layer_image, grow are generated according to their shape.
  • invert_mask: Whether to reverse the mask.
  • blend_mode3: Blending mode of glow.
  • opacity: Opacity of glow.
  • brightness: Luminance of light.
  • glow_range: range of glow.
  • blur:blur of glow.
  • light_color4: Center part color of glow.
  • glow_color4: Edge part color of glow.
  • note

Generate a stroke of layer。 image

Node options:
image

  • background_image1: The background image.
  • layer_image1: Layer image for composite.
  • layer_mask1,2: Mask for layer_image, stroke are generated according to their shape.
  • invert_mask: Whether to reverse the mask.
  • blend_mode3: Blending mode of stroke.
  • opacity: Opacity of stroke.
  • stroke_grow: Stroke expansion/contraction amplitude, positive values indicate expansion and negative values indicate contraction.
  • stroke_width: Stroke width.
  • blur: Blur of stroke.
  • stroke_color4: Stroke color, described in hexadecimal RGB format.
  • note

Generate gradient overlay image

Node options:

  • background_image1: The background image.
  • layer_image1: Layer image for composite.
  • layer_mask1,2: Mask for layer_image.
  • invert_mask: Whether to reverse the mask.
  • blend_mode3: Blending mode of gradient.
  • opacity: Opacity of stroke.
  • start_color: Color at the beginning of the gradient.
  • start_alpha: Transparency at the beginning of the gradient.
  • end_color: Color at the end of the gradient.
  • end_alpha: Transparency at the end of the gradient.
  • angle: Gradient rotation angle.
  • note

Generate color overlay image

Node options: image

  • background_image1: The background image.
  • layer_image1: Layer image for composite.
  • layer_mask1,2: Mask for layer_image.
  • invert_mask: Whether to reverse the mask.
  • blend_mode3: Blending mode of color.
  • opacity: Opacity of stroke.
  • color: Color of overlay.
  • note

image
image

LUT Apply

Apply LUT to the image. only supports .cube format. image

Node options:
image

  • LUT*: Here is a list of available. cube files in the LUT folder, and the selected LUT files will be applied to the image.
  • color_space: For regular image, please select linear, for image in the log color space, please select log.
  • strength: Range 0~100, LUT application strength. The larger the value, the greater the difference from the original image, and the smaller the value, the closer it is to the original image.

*LUT folder is defined in resource_dir.ini, this file is located in the root directory of the plug-in, and the default name is resource_dir.ini.example. to use this file for the first time, you need to change the file suffix to .ini. Open the text editing software and find the line starting with "LUT_dir=", after "=", enter the custom folder path name. support defining multiple folders in resource-dir.ini, separated by commas, semicolons, or spaces. all .cube files in this folder will be collected and displayed in the node list during ComfyUI initialization. If the folder set in ini is invalid, the LUT folder that comes with the plugin will be enabled.

Automatically adjust the brightness, contrast, and white balance of the image. Provide some manual adjustment options to compensate for the shortcomings of automatic adjustment. image

Node Options:
image

  • strength: Strength of adjust. The larger the value, the greater the difference from the original image.
  • brightness: Manual adjustment of brightness.
  • contrast: Manual adjustment of contrast.
  • saturation: Manual adjustment of saturation.
  • red: Manual adjustment of the red channel.
  • green: Manual adjustment of the green channel.
  • blue: Manual adjustment of the blue channel.

On the basis of AutoAdjust, add mask input and only calculate the content inside the mask for automatic color adjustment. Add multiple automatic adjustment modes. image

The following changes have been made based on AutoAdjust: image

  • mask: Optional mask input.
  • mode: Automatic adjustment mode. "RGB" automatically adjusts according to the three channels of RGB, "lum + sat"automatically adjusts according to luminance and saturation, "luminance" automatically adjusts according to luminance, "saturation" automatically adjusts according to saturation, and "mono" automatically adjusts according to grayscale and outputs monochrome.

Automatically adjust too dark or too bright image to moderate brightness, and support mask input. When mask input, only the content of the mask part is used as the data source of the automatic brightness. The output is still the whole adjusted image. image

Node options:
image

  • strength: Automatically adjust the intensity of the brightness. The larger the value, the more biased towards the middle value, the greater the difference from the original picture.
  • saturation: Color saturation. Changes in brightness usually result in changes in color saturation, where appropriate compensation can be adjusted.

Auto adjust the color tone of the image to resemble the reference image. image

Node options:
image

  • opacity: The opacity of an image after adjusting its color tone.

Change the exposure of the image. image

Color of Shadow & Highlight

Adjust the color of the dark and bright parts of the image. image

Node options:
image

  • image: The input image.
  • mask: Optional input. if there is input, only the colors within the mask range will be adjusted.
  • shadow_brightness: The brightness of the dark area.
  • shadow_saturation: The color saturation in the dark area.
  • shadow_hue: The color hue in the dark area.
  • shadow_level_offset: The offset of values in the dark area, where larger values bring more areas closer to the bright into the dark area.
  • shadow_range: The transitional range of the dark area.
  • highlight_brightness: The brightness of the highlight area.
  • highlight_saturation: The color saturation in the highlight area.
  • highlight_hue: The color hue in the highlight area.
  • highlight_level_offset: The offset of values in the highlight area, where larger values bring more areas closer to the dark into the highlight area.
  • highlight_range: The transitional range of the highlight area.

Node option:

  • exposure: Exposure value. Higher values indicate brighter image.

Color of Shadow HighlightV2

A replica of the Color of Shadow & Highlight node, with the "&" character removed from the node name to avoid ComfyUI workflow parsing errors.

image
Change the color temperature of the image.

Node Options:
image

  • temperature: Color temperature value. Range between-100 and 100. The higher the value, the higher the color temperature (bluer); The lower the color temperature, the lower the color temperature (yellowish).

image
Change the levels of image.

Node Options:
image

  • channel: Select the channel you want to adjust. Available in RGB, red, green, blue.
  • black_point*: Input black point value. Value range 0-255, default 0.
  • white_point*: Input white point value. Value range 0-255, default 255.
  • gray_point: Input grey point values. Value range 0.01-9.99, default 1.
  • output_black_point*: Output black point value. Value range 0-255, default 0.
  • output_white_point*: Output white point value. Value range 0-255, default 255.

*If the black_point or output_black_point value is greater than white_point or output_white_point, the two values are swapped, with the larger value used as white_point and the smaller value used as black_point.

image
Change the color balance of an image.

Node Options:
image

  • cyan_red: Cyan-Red balance. negative values are leaning cyan, positive values are leaning red.
  • magenta_green: Megenta-Green balance. negative values are leaning megenta, positive values are leaning green.
  • yellow_blue: Yellow-Blue balance. negative values are leaning yellow, positive values are leaning blue.

Change the gamma value of the image.

Node options:
image

  • gamma: Value of the Gamma.

Brightness & Contrast

Change the brightness, contrast, and saturation of the image.

Node options:
image

  • brightness: Value of brightness.
  • contrast: Value of contrast.
  • saturation: Value of saturation.

A replica of the Brightness & Contrast node, with the "&" character removed from the node name to avoid ComfyUI workflow parsing errors.

Adjust the RGB channels of the image.

Node options:
image

  • R: R channel.
  • G: G channel.
  • B: B channel.

Adjust the YUV channels of the image.

Node options:
image

  • Y: Y channel.
  • U: U channel.
  • V: V channel.

Adjust the LAB channels of the image.

Node options:
image

  • L: L channel.
  • A: A channel.
  • B: B channel.

Adjust the HSV channels of the image.

Node options:
image

  • H: H channel.
  • S: S channel.
  • V: V channel.

image

Used for compositing layers, allowing for compositing layer images of different sizes on the background image, and setting positions and transformations. multiple mixing modes are available for selection, and transparency can be set.

The node provide layer transformation_methods and anti_aliasing options. helps improve the quality of synthesized images.

The node provides mask output that can be used for subsequent workflows. image

Node options:
image

  • background_image: The background image.
  • layer_image5: Layer image for composite.
  • layer_mask2,5: Mask for layer_image.
  • invert_mask: Whether to reverse the mask.
  • blend_mode3: Blending mode.
  • opacity: Opacity of blend.
  • x_percent: Horizontal position of the layer on the background image, expressed as a percentage, with 0 on the far left and 100 on the far right. It can be less than 0 or more than 100, indicating that some of the layer's content is outside the screen.
  • y_percent: Vertical position of the layer on the background image, expressed as a percentage, with 0 on the top and 100 on the bottom. For example, setting it to 50 indicates vertical center, 20 indicates upper center, and 80 indicates lower center.
  • mirror: Mirror flipping. Provide two flipping modes, horizontal flipping and vertical flipping.
  • scale: Layer magnification, 1.0 represents the original size.
  • aspect_ratio: Layer aspect ratio. 1.0 is the original ratio, a value greater than this indicates elongation, and a value less than this indicates flattening.
  • rotate: Layer rotation degree.
  • Sampling methods for layer enlargement and rotation, including lanczos, bicubic, hamming, bilinear, box and nearest. Different sampling methods can affect the image quality and processing time of the synthesized image.
  • anti_aliasing: Anti aliasing, ranging from 0 to 16, the larger the value, the less obvious the aliasing. An excessively high value will significantly reduce the processing speed of the node.
  • note

Crop the image according to the mask range, and set the size of the surrounding borders to be retained. This node can be used in conjunction with the RestoreCropBox and ImageScaleRestore nodes to crop and modify upscale parts of image, and then paste them back in place. image

Node options:
image

  • image5: The input image.
  • mask_for_crop5: Mask of the image, it will automatically be cut according to the mask range.
  • invert_mask: Whether to reverse the mask.
  • detect: Detection method, min_bounding_rect is the minimum bounding rectangle of block shape, max_inscribed_rect is the maximum inscribed rectangle of block shape, and mask-area is the effective area for masking pixels.
  • top_reserve: Cut the top to preserve size.
  • bottom_reserve: Cut the bottom to preserve size.
  • left_reserve: Cut the left to preserve size.
  • right_reserve: Cut the right to preserve size.
  • note

Output:

  • croped_image: The image after crop.
  • croped_mask: The mask after crop.
  • crop_box: The trimmed box data is used when restoring the RestoreCropBox node.
  • box_preview: Preview image of cutting position, red represents the detected range, and green represents the cutting range after adding the reserved border.

The V2 upgraded version of CropByMask. Supports crop_box input, making it easy to cut layers of the same size.

The following changes have been made based on CropByMask: image

  • The input mask_for_crop reanme to mask
  • Add optional inputs to the crop_box. If there are inputs here, mask detection will be ignored and this data will be directly used for cropping.
  • Add the option round_to_multiple to round the trimming edge length multiple. For example, setting it to 8 will force the width and height to be multiples of 8.

Restore the cropped image to the original image by CropByMask.

Node options:
image

  • background_image: The original image before cutting.
  • croped_image5: The cropped image. If the middle is enlarged, the size needs to be restored before restoration.
  • croped_mask5: The cut mask.
  • crop_box: Box data during cutting.
  • invert_mask: Whether to reverse the mask.
  • note

Parsing the corp_box to x , y , width , height . image

Image scaling. when this node is used in pairs, the image can be automatically restored to its original size on the second node. image

Node options:
image

  • image5: The input image.
  • mask2,5: Mask of image.
  • original_size: Optional input, used to restore the image to its original size.
  • scale: Scale ratio. when the original_size have input, or scale_ by_longest_side is set to True, this setting will be ignored.
  • scale_by_longest_side: Allow scaling by long edge size.
  • longest_side: When the scale_by_longest_side is set to True, this will be used this value to the long edge of the image. when the original_size have input, this setting will be ignored.

Outputs:

  • image: The scaled image.
  • mask: If have mask input, the scaled mask will be output.
  • original_size: The original size data of the image is used for subsequent node recovery.
  • width: The output image's width.
  • height: The output image's height.

The V2 upgraded version of ImageScaleRestore.

Node options:
image
The following changes have been made based on ImageScaleRestore:

  • scale_by: Allow scaling by specified dimensions for long, short, width, height, or total pixels. When this option is set to by_scale, use the scale value, and for other options, use the scale_by_length value.
  • scale_by_length: The value here is used as scale_by to specify the length of the edge.

Scale the image or mask to the size of the reference image (or reference mask). image

Node options: image

  • scale_as*: Reference size. It can be an image or a mask.
  • image: Image to be scaled. this option is optional input. if there is no input, a black image will be output.
  • mask: Mask to be scaled. this option is optional input. if there is no input, a black mask will be output.
  • fit: Scale aspect ratio mode. when the width to height ratio of the original image does not match the scaled size, there are three modes to choose from, The letterbox mode retains the complete frame and fills in the blank spaces with black; The crop mode retains the complete short edge, and any excess of the long edge will be cut off; The fill mode does not maintain frame ratio and fills the screen with width and height.
  • method: Scaling sampling methods, including lanczos, bicubic, hamming, bilinear, box, and nearest.

*Only limited to input images and masks. forcing the integration of other types of inputs will result in node errors.

Outputs:

  • image: If there is an image input, the scaled image will be output.
  • mask: If there is a mask input, the scaled mask will be output.
  • original_size: The original size data of the image is used for subsequent node recovery.
  • width: The output image's width.
  • height: The output image's height.

Scale the image or mask by aspect ratio. the scaled size can be rounded to a multiple of 8 or 16, and can be scaled to the long side size. image

Node options:
image

  • aspect_ratio: Here are several common frame ratios provided. alternatively, you can choose "original" to keep original ratio or customize the ratio using "custom".
  • proportional_width: Proportional width. if the aspect ratio option is not "custom", this setting will be ignored.
  • proportional_height: Proportional height. if the aspect ratio option is not "custom", this setting will be ignored.
  • fit: Scale aspect ratio mode. when the width to height ratio of the original image does not match the scaled size, there are three modes to choose from, The letterbox mode retains the complete frame and fills in the blank spaces with black; The crop mode retains the complete short edge, and any excess of the long edge will be cut off; The fill mode does not maintain frame ratio and fills the screen with width and height.
  • method: Scaling sampling methods, including lanczos, bicubic, hamming, bilinear, box, and nearest.
  • round_to_multiple: Round multiples. for example, setting it to 8 will force the width and height to be multiples of 8.
  • scale_by_longest_side: Allow scaling by long edge size.
  • longest_side: When the scale_by_longest_side is set to True, this will be used this value to the long edge of the image. when the original_size have input, this setting will be ignored.

Outputs:

  • image: If have image input, the scaled image will be output.
  • mask: If have mask input, the scaled mask will be output.
  • original_size: The original size data of the image is used for subsequent node recovery.
  • width: The output image's width.
  • height: The output image's height.

V2 Upgraded Version of ImageScaleByAspectRatio

Node options:
image
The following changes have been made based on ImageScaleByAspectRatio:

  • scale_to_side: Allow scaling by specified dimensions for long, short, width, height, or total pixels.
  • scale_to_length: The numerical value here serves as the length of the specified edge or the total pixels (kilo pixels) for scale_to_side.
  • background_color4: The color of the background.

Inference the prompts based on the image. this node is repackage of the ComfyUI_VLM_nodes's UForm-Gen2 Qwen Node, thanks to the original author. Download model files from huggingface or Baidu Netdisk to ComfyUI/models/LLavacheckpoints/files_for_uform_gen2_qwen folder.

image

Node Options:

  • question: Prompt of UForm-Gen-QWen model.

Use Microsoft Phi 3.5 text and visual models for local inference. Can be used to generate prompt words, process prompt words, or infer prompt words from images. Running this model requires at least 16GB of video memory. Download model files from BaiduNetdisk or huggingface.co/microsoft/Phi-3.5-vision-instruct and huggingface.co/microsoft/Phi-3.5-mini-instruct and copy to ComfyUI\models\LLM folder. image

Node Options:
image

  • image: Optional input. The input image will serve as the input for Phi-3.5-vision-instruct.
  • model: Selectable to load Phi-3.5-vision-instruct or Phi-3.5-mini-instruct model. The default value of auto will automatically load the corresponding model based on whether there is image input.
  • device: Model loading device. Supports CPU and CUDA.
  • dtype: The model loading accuracy has three options: fp16, bf16, and fp32.
  • cache_model: Whether to cache the model.
  • system_prompt: The system prompt of Phi-3.5-mini-instruct.
  • user_prompt: User prompt words for LLM model.
  • do_sample: The do_Sample parameter of LLM defaults to True.
  • temperature: The temperature parameter of LLM defaults to 0.5.
  • max_new_tokens: The max_new_token parameter of LLM defaults to 512.

UserPrompt preset for generating SD text to image prompt words.

Node options:
image

  • template: Prompt word template. Currently, only the 'SD txt2img prompt' is available.
  • describe: Prompt word description. Enter a simple description here.
  • limit_word: Maximum length limit for output prompt words. For example, 200 means that the output text will be limited to 200 words.

UserPrompt preset used to replace a keyword in text with different content. This is not only a simple replacement, but also a logical sorting of the text based on the context of the prompt words to achieve the rationality of the output content.

Node options:
image

  • orig_prompt: Original prompt word input.
  • template: Prompt word template. Currently, only 'prompt replace word' is available.
  • exclude_word: Keywords that need to be excluded.
  • replace_with_word: That word will replace the exclude_word.

Inference the prompts based on the image. it can replace key word for the prompt. This node currently uses Google Gemini API as the backend service. Please ensure that the network environment can use Gemini normally. Please apply for your API key on Google AI Studio, And fill it in api_key.ini, this file is located in the root directory of the plug-in, and the default name is api_key.ini.example. to use this file for the first time, you need to change the file suffix to .ini. Open it using text editing software, fill in your API key after google_api_key= and save it. image

Node options:
image

  • api: The Api used. At present, there are two options "gemini-1. 5-flash" and "google-gemini".
  • token_limit: The maximum token limit for generating prompt words.
  • exclude_word: Keywords that need to be excluded.
  • replace_with_word: That word will replace the exclude_word.

Enter simple prompt words, output polished prompt words, and support inputting images as references, and support Chinese input. This node currently uses Google Gemini API as the backend service. Please ensure that the network environment can use Gemini normally. Please apply for your API key on Google AI Studio, And fill it in api_key.ini, this file is located in the root directory of the plug-in, and the default name is api_key.ini.example. to use this file for the first time, you need to change the file suffix to .ini. Open it using text editing software, fill in your API key after google_api_key= and save it. image

Node options:
image

  • image: Optional, input image as a reference for prompt words.
  • api: The Api used. At present, there are two options "gemini-1. 5-flash" and "google-gemini".
  • token_limit: The maximum token limit for generating prompt words.
  • discribe: Enter a simple description here. supports Chinese text input.

Use the Florence 2 model to infer prompt words. The code for this node section is fromyiwangsimple/florence_dw, thanks to the original author. *When using it for the first time, the model will be automatically downloaded. You can also download the model file from BaiduNetdisk to ComfyUI/models/florence2 folder. image

Node Options: image

  • florence2_model: Florence2 model input.
  • image: Image input.
  • task: Select the task for florence2.
  • text_input: Text input for florence2.
  • max_new_tokens: The maximum number of tokens for generating text.
  • num_beams: The number of beam searches that generate text.
  • do_sample: Whether to use text generated sampling.
  • fill_mask: Whether to use text marker mask filling.

Use the blip-vqa model for visual question answering. Part of the code for this node is referenced from celoron/ComfyUI-VisualQueryTemplate, thanks to the original author.
*Download model files from BaiduNetdisk or huggingface.co/Salesforce/blip-vqa-capfilt-large and huggingface.co/Salesforce/blip-vqa-base and copy to ComfyUI\models\VQA folder.

image

Node Options: image

  • image: The image input.
  • vqa_model: The vqa model input, it from LoadVQAModel node.
  • question: Task text input. A single question is enclosed in curly braces "{}", and the answer to the question will be replaced in its original position in the text output. Multiple questions can be defined using curly braces in a single Q&A. For example, for a picture of an item placed in a scene, the question is:"{object color} {object} on the {scene}".

Load the blip-vqa model.

Node Options: image

  • model: There are currently two models to choose from "blip-vqa-base" and "blip-vqa-capfilt-large".
  • precision: The model accuracy has two options: "fp16" and "fp32".
  • device: The model running device has two options: "cuda" and "cpu".

Shift the image. this node supports the output of displacement seam masks, making it convenient to create continuous textures. image

Node options:
image

  • image5: The input image.
  • mask2,5: The mask of image.
  • shift_x: Horizontal distance of shift.
  • shift_y: Vertical distance of shift.
  • cyclic: Is the part of displacement that is out of bounds cyclic.
  • background_color4: Background color. if cyclic is set to False, the setting here will be used as the background color.
  • border_mask_width: Border mask width.
  • border_mask_blur: Border mask blur.
  • note

A simple node for composit layer image and background image, multiple blend modes are available for option, and transparency can be set. image

Node options: image

  • background_image1: The background image.
  • layer_image1: Layer image for composite.
  • layer_mask1,2: Mask for layer_image.
  • invert_mask: Whether to reverse the mask.
  • blend_mode3: Blending mode.
  • opacity: Opacity of blend.
  • note

Display multiple images in one reel. Text annotations can be added to each image in the reel. By using the ImageReelComposite node, multiple reel can be combined into one image. image

Node Options:
image

  • image1: The first image. it must be input.
  • image2: The second image. optional input.
  • image3: The third image. optional input.
  • image4: The fourth image. optional input.
  • image1_text: Text annotation for the first image.
  • image2_text: Text annotation for the second image.
  • image3_text: Text annotation for the third image.
  • image4_text: Text annotation for the fourth image.
  • reel_height: The height of reel.
  • border: The border width of the image in the reel.

Output:

Combine multiple reel into one image.

Node Options: image

  • reel_1: The first reel. it must be input.
  • reel_2: The second reel. optional input.
  • reel_3: The third reel. optional input.
  • reel_4: The fourth reel. optional input.
  • font_file**: Here is a list of available font files in the font folder, and the selected font files will be used to generate images.
  • border: The border width of the reel.
  • color_theme: Theme color for the output image.
    *The font folder is defined in resource_dir.ini, this file is located in the root directory of the plug-in, and the default name is resource_dir.ini.example. to use this file for the first time, you need to change the file suffix to .ini. Open the text editing software and find the line starting with "FONT_dir=", after "=", enter the custom folder path name. support defining multiple folders in resource-dir.ini, separated by commas, semicolons, or spaces. all font files in this folder will be collected and displayed in the node list during ComfyUI initialization. If the folder set in ini is invalid, the font folder that comes with the plugin will be enabled.

Adjust image opacity image

Node option:

  • image5: Image input, supporting RGB and RGBA. if is RGB, the alpha channel of the entire image will be automatically added.
  • mask2,5 : Mask input.
  • invert_mask: Whether to reverse the mask.
  • opacity: Opacity of image.
  • note

Modify web extensions from mtb nodes. Select colors on the color palette and output RGB values, thanks to the original author. image

Node options:

  • mode: The output format is available in hexadecimal (HEX) and decimal (DEC).

Output type:

  • value: String format.

Output the color value as a single R, G, B three decimal values. Supports HEX and DEC formats for ColorPicker node output.

image

Node Options:

  • color_value: Supports hexadecimal (HEX) or decimal (DEC) color values and should be of string or tuple type. Forcing in other types will result in an error.

Output color values as individual decimal values of H, S, and V (maximum value of 255). Supports HEX and DEC formats for ColorPicker node output. image

Node Options:

  • color_value: Supports hexadecimal (HEX) or decimal (DEC) color values and should be of string or tuple type. Forcing in other types will result in an error.

Output grayscale values based on color values. Supports outputting 256 level and 100 level grayscale values. image

Node Options:

  • color_value: Supports hexadecimal (HEX) or decimal (DEC) color values and should be of string or tuple type. Forcing in other types will result in an error.

Outputs:

  • gray(256_level): 256 level grayscale value. Integer type, range 0~255.
  • gray(100_level): 100 level grayscale value. Integer type, range 0~100.

Obtain the main color or average color from the image and output RGB values. image

Node options: image

  • mode: There are two modes to choose from, with the main color and average color.

Output type:

  • RGB color in HEX: The RGB color described by hexadecimal RGB format, like '#FA3D86'.
  • HSV color in list: The HSV color described by python's list data format.

V2 upgrade of GetColorTone. You can specify the dominant or average color to get the body or background. image

The following changes have been made on the basis of GetColorTong: image

  • color_of: Provides 4 options, mask, entire, background, and subject, to select the color of the mask area, entire picture, background, or subject, respectively.
  • remove_background_method: There are two methods of background recognition: BiRefNet and RMBG V1.4.
  • invert_mask: Whether to reverse the mask.
  • mask_grow: Mask expansion. For subject, a larger value brings the obtained color closer to the color at the center of the body.

Output:

  • image: Solid color picture output, the size is the same as the input picture.
  • mask: Mask output.

Obtain the main color of the image. You can obtain 5 colors. image image

Node Options: image

  • image: The image input.
  • k_means_algorithm:K-Means algorithm options. "lloyd" is the standard K-Means algorithm, while "elkan" is the triangle inequality algorithm, suitable for larger images.

Outputs:

  • preview_image: 5 main color preview images.
  • color_1~color_5: Color value output. Output an RGB string in HEX format.

Output the most similar color name in the color palette based on the color value. image

Node Options: image

  • color: Color value input, in HEX format RGB string format.
  • palette: Color palette. xkcd includes 949 colors, css3 includes 147 colors, and html4 includes 16 colors.

Output:

  • color_name: Color name in string.

Extend the canvas image

Node options: image

  • invert_mask: Whether to reverse the mask.
  • top: Top extension value.
  • bottom: Bottom extension value.
  • left: Left extension value.
  • right: Right extension value.
  • color; Color of canvas.

V2 upgrade to ExtendCanvas.

Based on ExtendCanvas, color is modified to be a string type, and it supports external ColorPicker input, Support negative value input, it means image will be cropped. image

XY to Percent

image
Convert absolute coordinates to percentage coordinates.

image
Node options:

  • x: Value of X.
  • y: Value of Y.

image
This node is used to transform layer_image separately, which can change size, rotation, aspect ratio, and mirror flip without changing the image size.

image
Node options:

  • x: Value of X.
  • y: Value of Y.
  • mirror: Mirror flipping. Provide two flipping modes, horizontal flipping and vertical flipping.
  • scale: Layer magnification, 1.0 represents the original size.
  • aspect_ratio: Layer aspect ratio. 1.0 is the original ratio, a value greater than this indicates elongation, and a value less than this indicates flattening.
  • rotate: Layer rotation degree.
  • Sampling methods for layer enlargement and rotation, including lanczos, bicubic, hamming, bilinear, box and nearest. Different sampling methods can affect the image quality and processing time of the synthesized image.
  • anti_aliasing: Anti aliasing, ranging from 0 to 16, the larger the value, the less obvious the aliasing. An excessively high value will significantly reduce the processing speed of the node.

Similar to LayerImageTransform node, this node is used to transform the layer_mask separately, which can scale, rotate, change aspect ratio, and mirror flip without changing the mask size.

image
Node options:

  • x: Value of X.
  • y: Value of Y.
  • mirror: Mirror flipping. Provide two flipping modes, horizontal flipping and vertical flipping.
  • scale: Layer magnification, 1.0 represents the original size.
  • aspect_ratio: Layer aspect ratio. 1.0 is the original ratio, a value greater than this indicates elongation, and a value less than this indicates flattening.
  • rotate: Layer rotation degree.
  • Sampling methods for layer enlargement and rotation, including lanczos, bicubic, hamming, bilinear, box and nearest. Different sampling methods can affect the image quality and processing time of the synthesized image.
  • anti_aliasing: Anti aliasing, ranging from 0 to 16, the larger the value, the less obvious the aliasing. An excessively high value will significantly reduce the processing speed of the node.

image
Generate an image of a specified color and size.

image
Node options:

  • width: Width of the image.
  • height: Height of the image.
  • color4: Color of the image.

The V2 upgraded version of ColorImage.

image
The following changes have been made based on ColorImage:

  • size_as*: Input image or mask here to generate image according to its size. Note that this input takes priority over other size settings.
  • size**: Size preset. the preset can be customized by the user. if have size_as input, this option will be ignored.
  • custom_width: Image width. it valid when size is set to "custom". if have size_as input, this option will be ignored.
  • custom_height: Image height. it valid when size is set to "custom". if have size_as input, this option will be ignored.

*Only limited to input images and masks. forcing the integration of other types of inputs will result in node errors. **The preset size is defined in custom_size.ini, this file is located in the root directory of the plug-in, and the default name is custom_size.ini.example. to use this file for the first time, you need to change the file suffix to .ini. Open with text editing software. Each row represents a size, with the first value being width and the second being height, separated by a lowercase "x" in the middle. To avoid errors, please do not enter extra characters.

image
Generate an image with a specified size and color gradient.

image
Node options:

  • width: Width of the image.
  • height: Height of the image.
  • angle: Angle of gradient.
  • start_color4: Color of the begging.
  • end_color4: Color of the ending.

The V2 upgraded version of GradientImage.

image
The following changes have been made based on GradientImage:

  • size_as*: Input image or mask here to generate image according to its size. Note that this input takes priority over other size settings.
  • size**: Size preset. the preset can be customized by the user. if have size_as input, this option will be ignored.
  • custom_width: Image width. it valid when size is set to "custom". if have size_as input, this option will be ignored.
  • custom_height: Image height. it valid when size is set to "custom". if have size_as input, this option will be ignored.

*Only limited to input images and masks. forcing the integration of other types of inputs will result in node errors. **The preset size is defined in custom_size.ini, this file is located in the root directory of the plug-in, and the default name is custom_size.ini.example. to use this file for the first time, you need to change the file suffix to .ini. Open with text editing software. Each row represents a size, with the first value being width and the second being height, separated by a lowercase "x" in the middle. To avoid errors, please do not enter extra characters.

image
Rating bulk pictures and outputting top-ranked pictures. it used [ImageReward] (https://github.com/THUDM/ImageReward) for image scoring, thanks to the original authors.

image
Node options:

  • prompt: Optional input. Entering prompt here will be used as a basis to determine how well it matches the picture.
  • output_nun: Number of pictures outputted. This value should be less than the picture batch.

Outputs:

  • images: Bulk pictures output from high to low in order of rating.
  • obsolete_images: Knockout pictures. Also output in order of rating from high to low.

image
Generate simple typesetting images and masks from text. This node references some of the functionalities and code of ZHO-ZHO-ZHO/ComfyUI-Text_Image-Composite, thanks to the original author.

image
Node options:

  • size_as*: The input image or mask here will generate the output image and mask according to their size. this input takes priority over the width and height below.
  • font_file**: Here is a list of available font files in the font folder, and the selected font files will be used to generate images.
  • align: Alignment options. There are three options: center, left, and right.
  • char_per_line: The number of characters per line, any excess will be automatically wrapped.
  • leading: The leading space.
  • font_size: The size of font.
  • text_color: The color of text.
  • stroke_width: The width of stroke.
  • stroke_color: The color of stroke.
  • x_offset: The horizontal offset of the text position.
  • y_offset: The vertical offset of the text position.
  • width: Width of the image. If there is a size_as input, this setting will be ignored.
  • height: Height of the image. If there is a size_as input, this setting will be ignored.

*Only limited to input image and mask. forcing the integration of other types of inputs will result in node errors.

**The font folder is defined in resource_dir.ini, this file is located in the root directory of the plug-in, and the default name is resource_dir.ini.example. to use this file for the first time, you need to change the file suffix to .ini. Open the text editing software and find the line starting with "FONT_dir=", after "=", enter the custom folder path name. support defining multiple folders in resource-dir.ini, separated by commas, semicolons, or spaces. all font files in this folder will be collected and displayed in the node list during ComfyUI initialization. If the folder set in ini is invalid, the font folder that comes with the plugin will be enabled.

image
Generate images and masks from text. support for adjusting the spacing between words and lines, horizontal and vertical adjustments, it can set random changes in each character, including size and position.

image
Node options:

  • size_as*: The input image or mask here will generate the output image and mask according to their size. this input takes priority over the width and height below.
  • font_file**: Here is a list of available font files in the font folder, and the selected font files will be used to generate images.
  • spacing: Word spacing.this value is in pixels.
  • leading: Row leading.this value is in pixels.
  • horizontal_border: Side margin. If the text is horizontal, it is the left margin, and if it is vertical, it is the right margin. this value is represents a percentage, for example, 50 indicates that the starting point is located in the center on both sides.
  • vertical_border: Top margin. this value is represents a percentage, for example, 10 indicates that the starting point is located 10% away from the top.
  • scale: The overall size of the text. the initial size of text is automatically calculated based on the screen size and text content, with the longest row or column by default adapting to the image width or height. adjusting the value here will scale the text as a whole. this value is represents a percentage, for example, 60 represents scaling to 60%.
  • variation_range: The range of random changes in characters. when this value is greater than 0, the character will undergo random changes in size and position, and the larger the value, the greater the magnitude of the change.
  • variation_seed: The seed for randomly. fix this value to individual characters changes generated each time will not change.
  • layout: Text layout. there are horizontal and vertical options to choose from.
  • width: Width of the image. If there is a size_as input, this setting will be ignored.
  • height: Height of the image. If there is a size_as input, this setting will be ignored.
  • text_color: The color of text.
  • background_color4: The color of background.

*Only limited to input image and mask. forcing the integration of other types of inputs will result in node errors.

**The font folder is defined in resource_dir.ini, this file is located in the root directory of the plug-in, and the default name is resource_dir.ini.example. to use this file for the first time, you need to change the file suffix to .ini. Open the text editing software and find the line starting with "FONT_dir=", after "=", enter the custom folder path name. support defining multiple folders in resource-dir.ini, separated by commas, semicolons, or spaces. all font files in this folder will be collected and displayed in the node list during ComfyUI initialization. If the folder set in ini is invalid, the font folder that comes with the plugin will be enabled.

image

This node is merged from heshengtao. The PR modifies the scaling of the image text node based on the TextImage node. The font spacing follows the scaling, and the coordinates are no longer based on the top left corner of the text, but on the center point of the entire line of text. Thank you for the author's contribution.

image
Erase objects from the image based on the mask. this node is repackage of IOPaint, powered by state-of-the-art AI models, thanks to the original author.
It is have LaMa, LDM, ZITS,MAT, FcF, Manga models and the SPREAD method to erase. Please refer to the original link for the introduction of each model.
Please download the model files from lama models(BaiduNetdisk) or lama models(Google Drive) to ComfyUI/models/lama folder.

Node optons: image

  • lama_model: Choose a model or method.
  • device: After correctly installing Torch and Nvidia CUDA drivers, using cuda will significantly improve running speed.
  • invert_mask: Whether to reverse the mask.
  • grow: Positive values expand outward, while negative values contract inward.
  • blur: Blur the edge.

image
Split the image channel into individual images.

Node options: image

  • mode: Channel mode, include RGBA, YCbCr, LAB adn HSV.

image
Merge each channel image into one image.

Node options: image

  • mode: Channel mode, include RGBA, YCbCr, LAB adn HSV.

image
Remove the alpha channel from the image and convert it to RGB mode. you can choose to fill the background and set the background color.

Node options: image

  • RGBA_image: The input image supports RGBA or RGB modes.
  • mask: Optional input mask. If there is an input mask, it will be used first, ignoring the alpha that comes with RGBA_image.
  • fill_background: Whether to fill the background.
  • background_color4: Color of background.

image
Merge the image and mask into an RGBA mode image containing an alpha channel.

image
Automatically cutout and crop the image according to the mask. it can specify the background color, aspect ratio, and size for output image. this node is designed to generate the image materials for training models.
*Please refer to the model installation methods for SegmentAnythingUltra and RemBgUltra.

Node options: image

  • background_color4: The background color.
  • aspect_ratio: Here are several common frame ratios provided. alternatively, you can choose "original" to keep original ratio or customize the ratio using "custom".
  • proportional_width: Proportional width. if the aspect ratio option is not "custom", this setting will be ignored.
  • proportional_height: Proportional height. if the aspect ratio option is not "custom", this setting will be ignored.
  • scale_by_longest_side: Allow scaling by long edge size.
  • longest_side: When the scale_by_longest_side is set to True, this will be used this value to the long edge of the image. when the original_size have input, this setting will be ignored.
  • detect: Detection method, min_bounding_rect is the minimum bounding rectangle, max_inscribed_rect is the maximum inscribed rectangle.
  • border_reserve: Keep the border. expand the cutting range beyond the detected mask body area.
  • ultra_detail_range: Mask edge ultra fine processing range, 0 is not processed, which can save generation time.
  • matting_method: The method of generate masks. There are two methods available: Segment Anything and RMBG 1.4. RMBG 1.4 runs faster.
  • sam_model: Select the SAM model used by Segment Anything here.
  • grounding_dino_model: Select the Grounding_Dino model used by Segment Anything here.
  • sam_threshold: The threshold for Segment Anything.
  • sam_prompt: The prompt for Segment Anything.

Output: cropped_image: Crop and replace the background image. box_preview: Crop position preview. cropped_mask: Cropped mask.

The V2 upgrad version of ImageAutoCrop, it has made the following changes based on the previous version: image

  • Add optional input for mask. when there is a mask input, use that input directly to skip the built-in mask generation.
  • Add fill_background. When set to False, the background will not be processed and any parts beyond the frame will not be included in the output range.
  • aspect_ratio adds the original option.
  • scale_by: Allow scaling by specified dimensions for longest, shortest, width, or height.
  • scale_by_length: The value here is used as scale_by to specify the length of the edge.

Automatically crop the image to the specified size. You can input a mask to preserve the specified area of the mask. This node is designed to generate image materials for training the model.

Node Options: image

  • image: The input image.
  • mask: Optional input mask. The masking part will be preserved within the range of the cutting aspect ratio.
  • aspect_ratio: The aspect ratio of the output. Here are common frame ratios provided, with "custom" being the custom ratio and "original" being the original frame ratio.
  • proportional_width: Proportionally wide. If the aspect_ratio option is not 'custom', this setting will be ignored.
  • proportional_height: High proportion. If the aspect_ratio option is not 'custom', this setting will be ignored.
  • method: Scaling sampling methods include Lanczos, Bicubic, Hamming, Bilinear, Box, and Nearest.
  • scale_to_side: Allow scaling to be specified by long side, short side, width, height, or total pixels.
  • scale_to_length: The value here is used as the scale_to-side to specify the length of the edge or the total number of pixels (kilo pixels).
  • round_to_multiple: Multiply to the nearest whole. For example, if set to 8, the width and height will be forcibly set to multiples of 8.

Outputs: cropped_image: The cropped image. box_preview: Preview of cutting position.

Using low frequency filtering and retaining high frequency to recover image details. Compared to kijai's DetailTransfer, this node is better integrated with the environment while retaining details. image

Node Options:
image

  • image: Background image input.
  • detail_image: Detail image input.
  • mask: Optional input, if there is a mask input, only the details of the mask part are restored.
  • keep_high_freq: Reserved range of high frequency parts. The larger the value, the richer the retained high-frequency details.
  • erase_low_freq: The range of low frequency parts of the erasure. The larger the value, the more the low frequency range of the erasure.
  • mask_blur: Mask edge blur. Valid only if there is masked input.

image
Obtain the width and height of the image.

Output:

  • width: The width of image.
  • height: The height of image.
  • original_size: The original size data of the image is used for subsequent node recovery.

Switch output from multiple input images and masks, supporting 9 sets of inputs. All input items are optional. if there is only image or mask in a set of input, the missing item will be output as None. image

Node options:
image

  • output: Switch output. the value is the corresponding input group. when the random-output option is True, this setting will be ignored.
  • random_output: When this is true, the output setting will be ignored and a random set will be output among all valid inputs.

Retrieve specified images or masks from batch images or masks. image

Node Options:
image

  • images: Batch images input. This input is optional.
  • masks: Batch masks input. This input is optional.
  • select: Select the output image or mask at the batch index value, where 0 is the first image. Multiple values can be entered, separated by any non numeric character, including but not limited to commas, periods, semicolons, spaces or letters, and even Chinese characters. Note: If the value exceeds the batch size, the last image will be output. If there is no corresponding input, an empty 64x64 image or a 64x64 black mask will be output.

image
Combine multiple paragraphs of text into one.

image
Used to provide assistance for workflow debugging. When running, the properties of any object connected to this node will be printed to the console.

This node allows any type of input.

image
Output a string.

image
Output a string. same as TextBox.

image
Output a integer value.

image
Output a floating-point value with a precision of 5 decimal places.

image
Output a boolean value.

Used to generate random value within a specified range, with outputs of int, float, and boolean. Supports batch and list generation, and supports batch generation of a set of different random number lists based on image batch. image

Node Options: image

  • image: Optional input, generate a list of random numbers that match the quantity in batches according to the image.
  • min_value: Minimum value. Random numbers will randomly take values from the minimum to the maximum.
  • max_value: Maximum value. Random numbers will randomly take values from the minimum to the maximum.
  • float_decimal_places: Precision of float value.
  • fix_seed:Is the random number seed fixed. If this option is fixed, the generated random number will always be the same.

Outputs: int: Integer random number. float: Float random number. bool: Boolean random number.

image
Performs mathematical operations on two numeric values and outputs integer and floating point results*. Supported operations include+, -, *, /, **, //, %.

* The input only supports boolean, integer, and floating point numbers, forcing in other data will result in error.

image
The upgraded version of NumberCalculator has added numerical inputs within nodes and square root operations. The square root operation option is nth_root Note: The input takes priority, and when there is input, the values within the node will be invalid.

image
Perform a Boolean operation on two numeric values and output the result*. Supported operations include==, !=, and, or, xor, not, min, max.

* The input only supports boolean, integer, and floating point numbers, forcing in other data will result in error. The and operation between the values outputs a larger number, and the or operation outputs a smaller number.

image
The upgraded version of Boolean Operator has added numerical inputs within nodes and added judgments for greater than, less than, greater than or equal to, and less than or equal to. Note: The input takes priority, and when there is input, the values within the node will be invalid.

image
Determine whether the text contains or does not contain substrings, and output a Boolean value.

Node Options:
image

  • text: Input text.
  • condition: Judgment conditions. include determines whether it contains a substring, and exclude determines whether it does not.
  • sub_string: Substring.

Check if the mask contains enough valid areas and output a Boolean value.

Node Options:
image

  • white_point: The white point threshold used to determine whether the mask is valid is considered valid if it exceeds this value.
  • area_percent: The percentage of effective areas. If the proportion of effective areas exceeds this value, output True.

On the basis of CheckMask, the method option has been added, which allows for the selection of different detection methods. The area_percent is changed to a floating point number with an accuracy of 2 decimal places, which can detect smaller effective areas.

Node Options:
image

  • method: There are two detection methods, which are simple and detectability. The simple method only detects whether the mask is completely black, while the detect_percent method detects the proportion of effective areas.

image
Switches output based on Boolean conditional input. It can be used for any type of data switching, including but not limited to numeric values, strings, pictures, masks, models, latent, pipe pipelines, etc.

Node Options:
image

  • if_condition: Conditional input. Boolean, integer, floating point, and string inputs are supported. When entering a value, 0 is judged to be False; When a string is entered, an empty string is judged as Flase.
  • when_True: This item is output when the condition is True.
  • when_False: This item is output when the condition is False.

image
Switches the output based on the matching string. It can be used for any type of data switching, including but not limited to numeric values, strings, pictures, masks, models, latent, pipe pipelines, etc. Supports up to 3 sets of case switches. Compare case to switch_condition , if the same, output the corresponding input. If there are the same cases, the output is prioritized in order. If there is no matching case, the default input is output. Note that the string is case sensitive and Chinese and English full-width and half-width.

Node Options:
image

  • input_default: Input entry for default output. This input is required.
  • input_1: Input entry used to match case_1. This input is optional.
  • input_2: Input entry used to match case_2. This input is optional.
  • input_3: Input entry used to match case_3. This input is optional.
  • switch_condition: String used to judge with case.
  • case_1: case_1 string.
  • case_2: case_2 string.
  • case_3: case_3 string.

image
Stop the current queue. When executed at this node, the queue will stop. The workflow diagram above illustrates that if the image is larger than 1Mega pixels, the queue will stop executing.

Node Options:
image

  • mode: Stop mode. If you choose stop, it will be determined whether to stop based on the input conditions. If you choose continue, ignore the condition to continue executing the queue.
  • stop: If true, the queue will stop. If false, the queue will continue to execute.

image
Clean up GPU VRAM and system RAM. any type of input can be accessed, and when executed to this node, the VRAM and garbage objects in the RAM will be cleaned up. Usually placed after the node where the inference task is completed, such as the VAE Decode node.

Node Options:

  • purge_cache: Clean up cache。
  • purge_models: Unload all loaded models。

image
Enhanced save image node. You can customize the directory where the picture is saved, add a timestamp to the file name, select the save format, set the image compression rate, set whether to save the workflow, and optionally add invisible watermarks to the picture. (Add information in a way that is invisible to the naked eye, and use the ShowBlindWaterMark node to decode the watermark). Optionally output the json file of the workflow.

Node Options: image

  • iamge: The input image.
  • custom_path*: User-defined directory, enter the directory name in the correct format. If empty, it is saved in the default output directory of ComfyUI.
  • filename_prefix*: The prefix of file name.
  • timestamp: Timestamp the file name, opting for date, time to seconds, and time to milliseconds.
  • format: The format of image save. Currently available in png and jpg. Note that only png format is supported for RGBA mode pictures.
  • quality: Image quality, the value range 10-100, the higher the value, the better the picture quality, the volume of the file also correspondingly increases.
  • meta_data: Whether to save metadata to png file, that is workflow information. Set this to false if you do not want the workflow to be leaked.
  • blind_watermark: The text entered here (does not support multilingualism) will be converted into a QR code and saved as an invisible watermark. Use ShowBlindWaterMark node can decode watermarks. Note that pictures with watermarks are recommended to be saved in png format, and lower-quality jpg format will cause watermark information to be lost.
  • save_workflow_as_json: Whether the output workflow is a json file at the same time (the output json is in the same directory as the picture).
  • preview: Preview switch.

* Enter%date for the current date (YY-mm-dd) and %time for the current time (HH-MM-SS). You can enter / for subdirectories. For example, %date/name_%tiem will output the image to the YY-mm-dd folder, with name_HH-MM-SS as the file name prefix.

image
The node used to save the training set images and their text labels, where the image files and text label files have the same file name. Customizable directory for saving images, adding timestamps to file names, selecting save formats, and setting image compression rates. *The workflow image_tagger_stave.exe is located in the workflow directory.

Node Options: image

  • iamge: The input image.
  • tag_text: Text label of image.
  • custom_path*: User-defined directory, enter the directory name in the correct format. If empty, it is saved in the default output directory of ComfyUI.
  • filename_prefix*: The prefix of file name.
  • timestamp: Timestamp the file name, opting for date, time to seconds, and time to milliseconds.
  • format: The format of image save. Currently available in png and jpg. Note that only png format is supported for RGBA mode pictures.
  • quality: Image quality, the value range 10-100, the higher the value, the better the picture quality, the volume of the file also correspondingly increases.
  • preview: Preview switch.

* Enter%date for the current date (YY-mm-dd) and %time for the current time (HH-MM-SS). You can enter / for subdirectories. For example, %date/name_%tiem will output the image to the YY-mm-dd folder, with name_HH-MM-SS as the file name prefix.

image
Add an invisible watermark to a picture. Add the watermark image in a way that is invisible to the naked eye, and use the ShowBlindWaterMark node to decode the watermark.

Node Options: image

  • iamge: The input image.
  • watermark_image: Watermark image. The image entered here will automatically be converted to a square black and white image as a watermark. It is recommended to use a QR code as a watermark.

Decoding the invisible watermark added to the AddBlindWaterMark and SaveImagePlus nodes. image

Generate a square QR code picture.

Node Options:
image

  • size: The side length of image.
  • border: The size of the border around the QR code, the larger the value, the wider the border.
  • text: Enter the text content of the QR code here, and multi-language is not supported.

Decoding the QR code.

Node Options:
image

  • image: The input QR code image.
  • pre_blur: Pre-blurring, you can try to adjust this value for QR codes that are difficult to identify.

image
image
Load the PSD format file and export the layers. Note that this node requires the installation of the psd_tools dependency package, If error occurs during the installation of psd_tool, such as ModuleNotFoundError: No module named 'docopt' , please download docopt's whl and manual install it.

Node Options:
image

  • image: Here is a list of *.psd files under ComfyUI/input, where previously loaded psd images can be selected.
  • file_path: The complete path and file name of the psd file.
  • include_hidden_layer: whether include hidden layers.
  • find_layer_by: The method for finding layers can be selected by layer key number or layer name. Layer groups are treated as one layer.
  • layer_index: The layer key number, where 0 is the bottom layer, is incremented sequentially. If include_hiddenlayer is set to false, hidden layers are not counted. Set to -1 to output the top layer.
  • layer_name: Layer name. Note that capitalization and punctuation must match exactly.

Outputs: flat_image: PSD preview image. layer_iamge: Find the layer output. all_layers: Batch images containing all layers.

image
Encapsulate the four nodes of Negative Condition in SD3 into a separate node.

Node Options:
image

  • zero_out_start: Set the ConditioningSetTimestepRange start value for Negative ConditioningZeroOut, which is the same as the ConditioningSetTimestepRange end value for Negative.

image

Reproduction of Photoshop's layer Style - Blend If function. This node outputs a mask for layer composition on the ImageBlend or ImageBlendAdvance nodes. mask is an optional input, and if you enter a mask here, it will act on the output. image

Node Options:
image

  • invert_mask: Whether to reverse the mask.
  • blend_if: Channel selection for Blend If. There are four options: gray , red, green, and blue.
  • black_point: Black point values, ranging from 0-255.
  • black_range: Dark part transition range. The larger the value, the richer the transition level of the dark part mask.
  • white_point: White point values, ranging from 0-255.
  • white_range: Brightness transition range. The larger the value is, the richer the transition level of the bright part mask is.

Detect the area where the mask is located and output its position and size. image

Node options:
image

  • detect: Detection method, min_bounding_rect is the minimum bounding rectangle of block shape, max_inscribed_rect is the maximum inscribed rectangle of block shape, and mask-area is the effective area for masking pixels.
  • x_adjust: Adjust of horizontal deviation after detection.
  • y_adjust: Adjust of vertical offset after detection.
  • scale_adjust: Adjust the scaling offset after detection.

Output:

  • box_preview: Preview image of detection results. Red represents the detected result, and green represents the adjust output result.
  • x_percent: Horizontal position output in percentage.
  • y_percent: Vertical position output in percentage.
  • width: Width.
  • height: Height.
  • x: The x-coordinate of the top left corner position.
  • y: The y-coordinate of the top left corner position.

Ultra Nodes

image
Nodes that use ultra fine edge masking processing methods, the latest version of nodes includes: SegmentAnythingUltraV2, RmBgUltraV2, BiRefNetUltra, PersonMaskUltraV2, SegformerB2ClothesUltra and MaskEdgeUltraDetailV2. There are three edge processing methods for these nodes:

  • PyMatting optimizes the edges of the mask by using a closed form matching to mask trimap.
  • GuideFilter uses opencv guidedfilter to feather edges based on color similarity, and performs best when edges have strong color separation.
    The code for the above two methods is from the ComfyUI-Image-Filters in spacepxl's Alpha Matte, thanks to the original author.
  • VitMatte uses the transformer vit model for high-quality edge processing, preserving edge details and even generating semi transparent masks. Note: When running for the first time, you need to download the vitmate model file and wait for the automatic download to complete. If the download cannot be completed, you can run the command huggingface-cli download hustvl/vitmatte-small-composition-1k to manually download. After successfully downloading the model, you can use VITMatte(local) without accessing the network.
  • VitMatte's options: device set whether to use CUDA for vitimate operations, which is about 5 times faster than CPU. max_megapixels set the maximum image size for vitmate operation, and oversized images will be reduced in size. For 16G VRAM, it is recommended to set it to 3.

*Download all model files from BaiduNetdisk or Huggingface to ComfyUI/models/vitmatte folder.

The following figure is an example of the difference in output between three methods. image

Improvements to ComfyUI Segment Anything, thanks to the original author.

*Please refer to the installation of ComfyUI Segment Anything to install the model. If ComfyUI Segment Anything has been correctly installed, you can skip this step.

Node options:
image

  • sam_model: Select the SAM model.
  • ground_dino_model: Select the Grounding DINO model.
  • threshold: The threshold of SAM.
  • detail_range: Edge detail range.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.
  • prompt: Input for SAM's prompt.

The V2 upgraded version of SegmentAnythingUltra has added the VITMatte edge processing method.(Note: Images larger than 2K in size using this method will consume huge memory) image

On the basis of SegmentAnythingUltra, the following changes have been made: image

  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

This node is modified from kijai/ComfyUI-segment-anything-2. Thank to kijai for making significant contributions to the Comfyui community.
SAM2 Ultra node only support single image. If you need to process multiple images, please first convert the image batch to image list.
*Download models from BaiduNetdisk or huggingface.co/Kijai/sam2-safetensors and copy to ComfyUI/models/sam2 folder.

image

Node Options:
image

  • image: The image to segment.
  • bboxes: Input recognition box data.
  • sam2_model: Select the SAM2 model.
  • presicion: Model's persicion. can be selected from fp16, bf16, and fp32.
  • bbox_select: Select the input box data. There are three options: "all" to select all, "first" to select the box with the highest confidence, and "by_index" to specify the index of the box.
  • select_index: This option is valid when bbox_delect is 'by_index'. 0 is the first one. Multiple values can be entered, separated by any non numeric character, including but not limited to commas, periods, semicolons, spaces or letters, and even Chinese.
  • cache_model: Whether to cache the model. After caching the model, it will save time for model loading.
  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

SAM2 Video Ultra node support processing multiple frames of images or video sequences. Please define the recognition box data in the first frame of the sequence to ensure correct recognition.

sam2_video_ultra_example.mp4
2024-09-03.152625.mp4

Node Options:
image

  • image: The image to segment.
  • bboxes: Optional input of recognition bbox data. bboxes and first_frame_mask must have least one input. If first_frame_mask inputed, bbboxes will be ignored.
  • first_frame_mask: Optional input of the first frame mask. The mask will be used as the first frame recognition object. bboxes and first_frame_mask must have least one input. If first_frame_mask inputed, bbboxes will be ignored.
  • pre_mask: Optional input mask, which will serve as a propagation focus range limitation and help improve recognition accuracy.
  • sam2_model: Select the SAM2 model.
  • presicion: Model's persicion. can be selected from fp16 and bf16.
  • cache_model: Whether to cache the model. After caching the model, it will save time for model loading.
  • individual_object: When set to True, it will focus on identifying a single object. When set to False, attempts will be made to generate recognition boxes for multiple objects.
  • mask_preview_color: Display the color of non masked areas in the preview output.
  • detail_method: Edge processing methods. Only VITMatte method can be used.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.
  • device: Only cuda can be used.
  • max_megapixels: Set the maximum size for VitMate operations.A larger size will result in finer mask edges, but it will lead to a significant decrease in computation speed.

Use the Florence2 model to identify objects in images and output recognition box data.
*Download models from BaiduNetdisk and copy to ComfyUI/models/florence2 folder.

Node Options:
image

  • image: The image to segment.
  • florence2_model: Florence2 model, it from LoadFlorence2Model node.
  • prompt: Describe the object that needs to be identified.
  • sort_method: The selection box sorting method has 4 options: "left_to_right", "top_to_bottom", "big_to_small" and "confidence".
  • bbox_select: Select the input box data. There are three options: "all" to select all, "first" to select the box with the highest confidence, and "by_index" to specify the index of the box.
  • select_index: This option is valid when bbox_delect is 'by_index'. 0 is the first one. Multiple values can be entered, separated by any non numeric character, including but not limited to commas, periods, semicolons, spaces or letters, and even Chinese.

Use the YOLO-World model to identify objects in images and output recognition box data.
*Download models from BaiduNetdisk or GoogleDrive and copy to ComfyUI/models/yolo-world folder.

Node Options:
image

  • image: The image to segment.
  • confidence_threshold: The threshold of confidence.
  • nms_iou_threshold: The threshold of Non-Maximum Suppression.
  • prompt: Describe the object that needs to be identified.
  • sort_method: The selection box sorting method has 4 options: "left_to_right", "top_to_bottom", "big_to_small" and "confidence".
  • bbox_select: Select the input box data. There are three options: "all" to select all, "first" to select the box with the highest confidence, and "by_index" to specify the index of the box.
  • select_index: This option is valid when bbox_delect is 'by_index'. 0 is the first one. Multiple values can be entered, separated by any non numeric character, including but not limited to commas, periods, semicolons, spaces or letters, and even Chinese.

Use the YOLO-8 model to identify objects in images and output recognition box data.
*Download models from GoogleDrive or BaiduNetdisk and copy to ComfyUI/models/yolo folder.

Node Options:
image

  • image: The image to segment.
  • yolo_model: Choose the yolo model.
  • sort_method: The selection box sorting method has 4 options: "left_to_right", "top_to_bottom", "big_to_small" and "confidence".
  • bbox_select: Select the input box data. There are three options: "all" to select all, "first" to select the box with the highest confidence, and "by_index" to specify the index of the box.
  • select_index: This option is valid when bbox_delect is 'by_index'. 0 is the first one. Multiple values can be entered, separated by any non numeric character, including but not limited to commas, periods, semicolons, spaces or letters, and even Chinese.

Use mask as recognition box data. All areas surrounded by white areas on the mask will be recognized as an object. Multiple enclosed areas will be identified separately.

Node Options:
image

  • object_mask: The mask input.
  • sort_method: The selection box sorting method has 4 options: "left_to_right", "top_to_bottom", "big_to_small" and "confidence".
  • bbox_select: Select the input box data. There are three options: "all" to select all, "first" to select the box with the highest confidence, and "by_index" to specify the index of the box.
  • select_index: This option is valid when bbox_delect is 'by_index'. 0 is the first one. Multiple values can be entered, separated by any non numeric character, including but not limited to commas, periods, semicolons, spaces or letters, and even Chinese.

Merge recognition box data.

Node Options:
image

  • bboxes_1: Required input. The first set of identification boxes.
  • bboxes_2: Optional input. The second set of identification boxes.
  • bboxes_3: Optional input. The third set of identification boxes.
  • bboxes_4: Optional input. The fourth set of identification boxes.

Draw the recognition BBoxes data output by the Object Detector node as a mask.
image

Node Options:
image

  • image: Image input. It must be consistent with the image recognized by the Object Detector node.
  • bboxes: Input recognition BBoxes data.
  • grow_top: Each BBox expands upwards as a percentage of its height, positive values indicate upward expansion and negative values indicate downward expansion.
  • grow_bottom: Each BBox expands downwards as a percentage of its height, positive values indicating downward expansion and negative values indicating upward expansion.
  • grow_left: Each BBox expands to the left as a percentage of its width, positive values expand to the left and negative values expand to the right.
  • grow_right: Each BBox expands to the right as a percentage of its width, positive values indicate expansion to the right and negative values indicate expansion to the left.

This node is implementation of EVF-SAM in ComfyUI.
*Please download model files from BaiduNetdisk or huggingface/EVF-SAM2, huggingface/EVF-SAM to ComfyUI/models/EVF-SAM folder(save the models in their respective subdirectories). image

Node Options:
image

  • image: The input image.
  • model: Select the model. Currently, there are options for evf-sam2 and evf sam.
  • presicion: Model accuracy can be selected from fp16, bf16, and fp32.
  • load_in_bit: Load the model with positional accuracy. You can choose from full, 8, and 4.
  • pormpt: Prompt words used for segmentation.
  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

Using the segmentation function of the Florence2 model, while also having ultra-high edge details. The code for this node section is from spacepxl/ComfyUI-Florence-2, thanks to the original author. *Download the model files from BaiduNetdisk to ComfyUI/models/florence2 folder.

image

Node Options:
image

  • florence2_model: Florence2 model input.
  • image: Image input.
  • task: Select the task for florence2.
  • text_input: Text input for florence2.
  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

Florence2 model loader. *When using it for the first time, the model will be automatically downloaded.

image
At present, there are base, base-ft, large, large-ft, DocVQA, SD3-Captioner and base-PromptGen models to choose from.

Remove background. compared to the similar background removal nodes, this node has ultra-high edge details.

This node combines the Alpha Matte node of Spacepxl's ComfyUI-Image-Filters and the functionality of ZHO-ZHO-ZHO's ComfyUI-BRIA_AI-RMBG, thanks to the original author.

*Download model files from BRIA Background Removal v1.4 or BaiduNetdisk to ComfyUI/models/rmbg/RMBG-1.4 folder. This model can be used for non-commercial purposes.

image

Node options:
image

  • detail_range: Edge detail range.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.

The V2 upgraded version of RemBgUltra has added the VITMatte edge processing method.(Note: Images larger than 2K in size using this method will consume huge memory)

On the basis of RemBgUltra, the following changes have been made: image

  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

Using the BiRefNet model to remove background has better recognition ability and ultra-high edge details. The code for the model part of this node comes from Viper's ComfyUI-BiRefNet,thanks to the original author.

*From https://huggingface.co/ViperYX/BiRefNet or BaiduNetdisk download the BiRefNet-ep480.pth,pvt_v2_b2.pth,pvt_v2_b5.pth,swin_base_patch4_window12_384_22kto1k.pth, swin_large_patch4_window12_384_22kto1k.pth 5 files to ComfyUI/models/BiRefNet folder.

image

Node options:
image

  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

This node supports the use of the latest BiRefNet model. *Download model file from BaiduNetdisk or GoogleDrive named BiRefNet-general-epoch_244.pth to ComfyUI/Models/BiRefNet/pth folder. You can also download more BiRefNet models and put them here.

image

Node Options:
image

  • image: The input image.
  • birefnet_model: The BiRefNet model is input and it is output from the LoadBiRefNetModel node.
  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Due to the excellent edge processing of BiRefNet, it is set to False by default here.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

Load the BiRefNet model.

Node Options:
image

  • model: Select the model. List the files in the CoomfyUI/models/BiRefNet/pth folder for selection.

Using the transparent-background model to remove background has better recognition ability and speed, while also having ultra-high edge details.

*From googledrive or BaiduNetdisk download all files to ComfyUI/models/transparent-background folder.

image

Node Options:
image

  • model: Select the model.
  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

Generate masks for portrait's face, hair, body skin, clothing, or accessories. Compared to the previous A Person Mask Generator node, this node has ultra-high edge details. The model code for this node comes from a-person-mask-generator, edge processing code from ComfyUI-Image-Filters,thanks to the original author. *Download model files from BaiduNetdisk to ComfyUI/models/mediapipe folder.

image

Node options:
image

  • face: Face recognition.
  • hair: Hair recognition.
  • body: Body skin recognition.
  • clothes: Clothing recognition.
  • accessories: Identification of accessories (such as backpacks).
  • background: Background recognition.
  • confidence: Recognition threshold, lower values will output more mask ranges.
  • detail_range: Edge detail range.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.

The V2 upgraded version of PersonMaskUltra has added the VITMatte edge processing method.(Note: Images larger than 2K in size using this method will consume huge memory)

On the basis of PersonMaskUltra, the following changes have been made: image

  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

image
Generate masks for characters' faces, hair, arms, legs, and clothing, mainly used for segmenting clothing. The model segmentation code is fromStartHua,thanks to the original author. Compared to the comfyui_segformer_b2_clothes, this node has ultra-high edge details. (Note: Generating images with edges exceeding 2K in size using the VITMatte method will consume a lot of memory)

*Download all model files from huggingface or BaiduNetdisk to ComfyUI/models/segformer_b2_clothes folder.

Node Options:
image

  • face: Facial recognition switch.
  • hair: Hair recognition switch.
  • hat: Hat recognition switch.
  • sunglass: Sunglass recognition switch.
  • left_arm: Left arm recognition switch.
  • right_arm: Right arm recognition switch.
  • left_leg: Left leg recognition switch.
  • right_leg: Right leg recognition switch.
  • skirt: Skirt recognition switch.
  • pants: Pants recognition switch.
  • dress: Dress recognition switch.
  • belt: Belt recognition switch.
  • shoe: Shoes recognition switch.
  • bag: Bag recognition switch.
  • scarf: Scarf recognition switch.
  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

image
image
Using the segformer model to segment clothing with ultra-high edge details. Currently supports segformer b2 clothes, segformer b3 clothes and segformer b3 fashion。

*Download modelfiles from huggingface or 百度网盘 to ComfyUI/models/segformer_b2_clothes folder.
*Download modelfiles from huggingface or 百度网盘 to ComfyUI/models/segformer_b3_clothes folder.
*Download modelfiles from huggingface or 百度网盘 to ComfyUI/models/segformer_b3_fashion folder.

Node Options:
image

  • image: The input image.
  • segformer_pipeline: Segformer pipeline input. The pipeline is output by SegformerClottesPipeline and SegformerFashionPipeline node.
  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

Select the segformer clothes model and choose the segmentation content.

Node Options:
image

  • model: Model selection. There are currently two models available to choose from for segformer b2 clothes and segformer b3 clothes.
  • face: Facial recognition switch.
  • hair: Hair recognition switch.
  • hat: Hat recognition switch.
  • sunglass: Sunglass recognition switch.
  • left_arm: Left arm recognition switch.
  • right_arm: Right arm recognition switch.
  • left_leg: Left leg recognition switch.
  • right_leg: Right leg recognition switch.
  • left_shoe: Left shoe recognition switch.
  • right_shoe: Right shoe recognition switch.
  • skirt: Skirt recognition switch.
  • pants: Pants recognition switch.
  • dress: Dress recognition switch.
  • belt: Belt recognition switch.
  • bag: Bag recognition switch.
  • scarf: Scarf recognition switch.

Select the segformer fashion model and choose the segmentation content.

Node Options:
image

  • model: Model selection. Currently, there is only one model available for selection: segformer b3 fashion。
  • shirt: shirt and blouse switch.
  • top: top, t-shirt, sweatshirt switch.
  • sweater: sweater switch.
  • cardigan: cardigan switch.
  • jacket: jacket switch.
  • vest: vest switch.
  • pants: pants switch.
  • shorts: shorts switch.
  • skirt: skirt switch.
  • coat: coat switch.
  • dress: dress switch.
  • jumpsuit: jumpsuit switch.
  • cape: cape switch.
  • glasses: glasses switch.
  • hat: hat switch.
  • hairaccessory: headband, head covering, hair accessory switch.
  • tie: tie switch.
  • glove: glove switch.
  • watch: watch switch.
  • belt: belt switch.
  • legwarmer: leg warmer switch.
  • tights: tights and stockings switch.
  • sock: sock switch.
  • shoe: shoes switch.
  • bagwallet: bag and wallet switch.
  • scarf: scarf switch.
  • umbrella: umbrella switch.
  • hood: hood switch.
  • collar: collar switch.
  • lapel: lapel switch.
  • epaulette: epaulette switch.
  • sleeve: sleeve switch.
  • pocket: pocket switch.
  • neckline: neckline switch.
  • buckle: buckle switch.
  • zipper: zipper switch.
  • applique: applique switch.
  • bead: bead switch.
  • bow: bow switch.
  • flower: flower switch.
  • fringe: fringe switch.
  • ribbon: ribbon switch.
  • rivet: rivet switch.
  • ruffle: ruffle switch.
  • sequin: sequin switch.
  • tassel: tassel switch.

Used for generate human body parts masks, it is based on the warrper of metal3d/ComfyUI_Human_Parts, thank the original author. This node has added ultra-fine edge processing based on the original work. Download model file from BaiduNetdisk or huggingface and copy to ComfyUI\models\onnx\human-parts folder. image

Node Options:
image

  • image: The input image.
  • face: Recognize face switch.
  • hair: Recognize hair switch.
  • galsses: Recognize glasses switch.
  • top_clothes: Recognize top clothes switch.
  • bottom_clothes: Recognize bottom clothes switch.
  • torso_skin: Recognize torso skin switch.
  • left_arm: Recognize left arm switch.
  • right_arm: Recognize right arm switch.
  • left_leg: Recognize left leg switch.
  • right_leg: Recognize right leg switch.
  • left_foot: Recognize left foot switch.
  • right_foot: Recognize right foot switch.
  • detail_method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • detail_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • detail_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.
  • process_detail: Set to false here will skip edge processing to save runtime.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

Process rough masks to ultra fine edges. This node combines the Alpha Matte and the Guided Filter Alpha nodes functions of Spacepxl's ComfyUI-Image-Filters, thanks to the original author. image

Node options:
image

  • method: Provide two methods for edge processing: PyMatting and OpenCV-GuidedFilter. PyMatching has a slower processing speed, but for video, it is recommended to use this method to obtain smoother mask sequences.
  • mask_grow: Mask expansion amplitude. positive values expand outward, while negative values contract inward. For rougher masks, negative values are usually used to shrink their edges for better results.
  • fix_gap: Repair the gaps in the mask. if obvious gaps in the mask, increase this value appropriately.
  • fix_threshold: The threshold of fix_gap.
  • detail_range: Edge detail range.
  • black_point: Edge black sampling threshold.
  • white_point: Edge white sampling threshold.

The V2 upgraded version of MaskEdgeUltraDetail has added the VITMatte edge processing method.(Note: Images larger than 2K in size using this method will consume huge memory)
This method is suitable for handling semi transparent areas.

On the basis of MaskEdgeUltraDetail, the following changes have been made: image

  • method: Edge processing methods. provides VITMatte, VITMatte(local), PyMatting, GuidedFilter. If the model has been downloaded after the first use of VITMatte, you can use VITMatte (local) afterwards.
  • edge_erode: Mask the erosion range inward from the edge. the larger the value, the larger the range of inward repair.
  • edge_dilate: The edge of the mask expands outward. the larger the value, the wider the range of outward repair.
  • device: Set whether the VitMatte to use cuda.
  • max_megapixels: Set the maximum size for VitMate operations.

Use the YoloV8 model to detect faces, hand box areas, or character segmentation. Supports the output of the selected number of channels. Download the model files from GoogleDrive or BaiduNetdisk to ComfyUI/models/yolo folder.

image

Node Options:
image

  • yolo_model: Yolo model selection. the model with seg name can output segmented masks, otherwise they can only output box masks.
  • mask_merge: Select the merged mask. all is to merge all mask outputs. The selected number is how many masks to output, sorted by recognition confidence to merge the output.

Outputs:

  • mask: The output mask.
  • yolo_plot_image: Preview of yolo recognition results.
  • yolo_masks: For all masks identified by yolo, each individual mask is output as a mask.

Use the Mediapipe model to detect facial features, segment left and right eyebrows, eyes, lips, and tooth. *Download the model files from BaiduNetdisk to ComfyUI/models/mediapipe folder.

image

Node Options:
image

  • left_eye: Recognition switch of left eye.
  • left_eyebrow: Recognition switch of left eyebrow.
  • right_eye: Recognition switch of right eye.
  • right_eyebrow: Recognition switch of right eyebrow.
  • lips: Recognition switch of lips.
  • tooth: Recognition switch of tooth.

Generate a mask based on the selected color. image

Node Options:
image

  • image: Input image.
  • mask: This input is optional, if there is a mask, only the colors inside the mask are included in the range.
  • color: Color selector. Click on the color block to select a color, and you can use the straws on the color picker panel to pick up the screen color. Note: When using straws, maximize the browser window.
  • color_in_HEX4: Enter color values. If this item has input, it will be used first, ignoring the color selected by the color .
  • threshold: Mask range threshold, the larger the value, the larger the mask range.
  • fix_gap: Repair the gaps in the mask. If there are obvious gaps in the mask, increase this value appropriately.
  • fix_threshold: The threshold for repairing masks.
  • invert_mask: Whether to reverse the mask.

Convert the image to a mask. Supports converting any channel in LAB, RGBA, YUV, and HSV modes into masks, while providing color scale adjustment. Support mask optional input to obtain masks that only include valid parts. image

Node Options:
image

  • image: Input image.
  • mask: This input is optional, if there is a mask, only the colors inside the mask are included in the range.
  • channel: Channel selection. You can choose any channel of LAB, RGBA, YUV, or HSV modes.
  • black_point*: Black dot value for the mask. The value range is 0-255, with a default value of 0.
  • white_point*: White dot value for the mask. The value range is 0-255, with a default value of 255.
  • gray_point: Gray dot values for the mask. The value range is 0.01-9.99, with a default of 1.
  • invert_output_mask: Whether to reverse the mask.

*If the black_point or output_black_point value is greater than white_point or output_white_point, the two values are swapped, with the larger value used as white_point and the smaller value used as black_point.

Shadow & Highlight Mask

Generate masks for the dark and bright parts of the image. image

Node options:
image

  • image: The input image.
  • mask: Optional input. if there is input, only the colors within the mask range will be adjusted.
  • shadow_level_offset: The offset of values in the dark area, where larger values bring more areas closer to the bright into the dark area.
  • shadow_range: The transitional range of the dark area.
  • highlight_level_offset: The offset of values in the highlight area, where larger values bring more areas closer to the dark into the highlight area.
  • highlight_range: The transitional range of the highlight area.

Shadow Highlight Mask V2

A replica of the Shadow & Highlight Mask node, with the "&" character removed from the node name to avoid ComfyUI workflow parsing errors.

Pixel expansion preprocessing on the masked edge of an image can effectively improve the edges of image composit. image

Node options:
image

  • invert_mask: Whether to reverse the mask.
  • mask_grow: Mask expansion amplitude.

Calculate the differences between two images and output them as mask. image

Node options:
image

  • gain: The gain of difference calculate. higher value will result in a more significant slight difference.
  • fix_gap: Fix the internal gaps of the mask. higher value will repair larger gaps.
  • fix_threshold: The threshold for fix_gap.
  • main_subject_detect: Setting this to True will enable subject detection, ignoring differences outside of the subject.

Grow and shrink edges and blur the mask image

Node options: image

  • invert_mask: Whether to reverse the mask.
  • grow: Positive values expand outward, while negative values contract inward.
  • blur: Blur the edge.

Smooth transition and shrink the mask edges while preserving edge details. image

Node options: image

  • invert_mask: Whether to reverse the mask.
  • shrink_level: Shrink the smoothness level.
  • soft: Smooth amplitude.
  • edge_shrink: Edge shrinkage amplitude.
  • edge_reserve: Preserve the amplitude of edge details, 100 represents complete preservation, and 0 represents no preservation at all.

Comparison of MaskGrow and MaskEdgeShrink image

Create motion blur on the mask. image

Node options:
image

  • invert_mask: Whether to reverse the mask.
  • blur: The size of blur.
  • angle: The angle of blur.

Create a gradient for the mask from one side. please note the difference between this node and the CreateGradientMask node. image

Node options: image

  • invert_mask: Whether to reverse the mask.
  • gradient_side: Generate gradient from which edge. There are four directions: top, bottom, left and right.
  • gradient_scale: Gradient distance. The default value of 100 indicates that one side of the gradient is completely transparent and the other side is completely opaque. The smaller the value, the shorter the distance from transparent to opaque.
  • gradient_offset: Gradient position offset.
  • opacity: The opacity of the gradient.

Create a gradient mask. please note the difference between this node and the MaskGradient node. image
image

Node options:
image

  • size_as*: The input image or mask here will generate the output image and mask according to their size. this input takes priority over the width and height below.
  • width: Width of the image. If there is a size_as input, this setting will be ignored.
  • height: Height of the image. If there is a size_as input, this setting will be ignored.
  • gradient_side: Generate gradient from which edge. There are five directions: top, bottom, left, right and center.
  • gradient_scale: Gradient distance. The default value of 100 indicates that one side of the gradient is completely transparent and the other side is completely opaque. The smaller the value, the shorter the distance from transparent to opaque.
  • gradient_offset: Gradient position offset. When gradient_side is center, the size of the gradient area is adjusted here, positive values are smaller, and negative values are enlarged.
  • opacity: The opacity of the gradient.

*Only limited to input image and mask. forcing the integration of other types of inputs will result in node errors.

Generate mask contour strokes. image

Node options:
image

  • invert_mask: Whether to reverse the mask.
  • stroke_grow: Stroke expansion/contraction amplitude, positive values indicate expansion and negative values indicate contraction.
  • stroke_width: Stroke width.
  • blur: Blur of stroke.

Generates noise for the mask. image

Node Options:
image

  • grain: Noise intensity.
  • invert_mask: Whether to reverse the mask.

Preview the input mask image

Invert the mask image

image

Sharp & Soft

Enhance or smooth out details for image. image

Node options: image

  • enhance: Provide 4 presets, which are very sharp, sharp, soft and very soft. If you choose None, you will not do any processing.

Make the skin look smoother. image

Node options: image

  • smooth: Skin smoothness.
  • threshold: Smooth range. the larger the range with the smaller value.
  • opacity: The opacity of the smoothness.

Watercolor painting effect image

Node option: image

  • line_density: The black line density.
  • opacity: The opacity of watercolor effects.

Soft light effect, the bright highlights on the screen appear blurry. image

Node options: image

  • soft: Size of soft light.
  • threshold: Soft light range. the light appears from the brightest part of the picture. in lower value, the range will be larger, and in higher value, the range will be smaller.
  • opacity: Opacity of the soft light.

Channel misalignment. similar to the effect of Tiktok logo. image

Node options: image

  • distance: Distance of channel separation.
  • angle: Angle of channel separation.
  • mode: Channel shift arrangement order.

HDR Effects

enhances the dynamic range and visual appeal of input images. This node is reorganize and encapsulate of HDR Effects (SuperBeasts.AI), thanks to the original author. image

Node options: image

  • hdr_intensity: Range: 0.0 to 5.0, Controls the overall intensity of the HDR effect, Higher values result in a more pronounced HDR effect.
  • shadow_intensity: Range: 0.0 to 1.0,Adjusts the intensity of shadows in the image,Higher values darken the shadows and increase contrast.
  • highlight_intensity: Range: 0.0 to 1.0,Adjusts the intensity of highlights in the image,Higher values brighten the highlights and increase contrast.
  • gamma_intensity: Range: 0.0 to 1.0,Controls the gamma correction applied to the image,Higher values increase the overall brightness and contrast.
  • contrast: Range: 0.0 to 1.0,Enhances the contrast of the image, Higher values result in more pronounced contrast.
  • enhance_color: Range: 0.0 to 1.0,Enhances the color saturation of the image, Higher values result in more vibrant colors.

Simulate the grain, dark edge, and blurred edge of the film, support input depth map to simulate defocus.
This node is reorganize and encapsulate of digitaljohn/comfyui-propost, thanks to the original author. image

Node options: image

  • image: The input image.
  • depth_map: Input depth map to simulate defocus effect. it is an optional input. if there is no input, will simulates radial blur at the edges of the image.
  • center_x: The horizontal axis of the center point position of the dark edge and radial blur, where 0 represents the leftmost side, 1 represents the rightmost side, and 0.5 represents at the center.
  • center_y: The vertical axis of the center point position of the dark edge and radial blur, where 0 represents the leftmost side, 1 represents the rightmost side, and 0.5 represents at the center.
  • saturation: Color saturation, 1 is the original value.
  • grain_power: Grain intensity. larger value means more pronounced the noise.
  • grain_scale: Grain size.
  • grain_sat: The color saturation of grain. 0 represents mono noise, and the larger the value, the more prominent the color.
  • grain_shadows: Grain intensity of dark part.
  • grain_highs: Grain intensity of light part.
  • blur_strength: The strength of blur. larger value means more blurry it becomes.
  • blur_focus_spread: Focus diffusion range. larger value means larger clear range.
  • focal_depth: Simulate the focal distance of defucus. 0 indicates that focus is farthest, and 1 indicates that is closest. this setting only valid when input the depth_map.

The upgraded version of the Film node adds the fastgrain method on the basis of the previous one, and the speed of generating noise is accelerated by 10 times. The code for fastgrain is from github.com/spacepxl/ComfyUI-Image-Filters BetterFilmGrain node, thanks to the original authors. image

Simulate the light leakage effect of the film. please download model file from Baidu Netdisk or [Google Drive]([light_leak.pkl(Google Drive)(https://drive.google.com/file/d/1DcH2Zkyj7W3OiAeeGpJk1eaZpdJwdCL-/view?usp=sharing)) and copy to ComfyUI/models/layerstyle folder. image

Node options: image

  • light: 32 types of light spots are provided. random is a random selection.
  • corner: There are four options for the corner where the light appears: top left, top right, bottom left, and bottom right.
  • hue: The hue of the light.
  • saturation: The color saturation of the light.
  • opacity: The opacity of the light.

Pseudo color heat map effect. image

Node options: image

  • color_map: Effect type. there are a total of 22 types of effects, as shown in the above figure.
  • opacity: The opacity of the color map effect.

Make the image motion blur image

Node options:

  • angle: The angle of blur.
  • blur: The size of blur.

Make the image gaussian blur image

Node options:

  • blur: The size of blur.

Add noise to the picture. image

Node Options: image

  • grain_power: Noise intensity.
  • grain_scale: Noise size.
  • grain_sat: Color saturation of noise.

Annotation for notes

1 The layer_image, layer_mask and the background_image(if have input), These three items must be of the same size.

2 The mask not a mandatory input item. the alpha channel of the image is used by default. If the image input does not include an alpha channel, the entire image's alpha channel will be automatically created. if have masks input simultaneously, the alpha channel will be overwrite by the mask.

3 The Blend Mode include normal, multply, screen, add, subtract, difference, darker, color_burn, color_dodge, linear_burn, linear_dodge, overlay, soft_light, hard_light, vivid_light, pin_light, linear_light, and hard_mix. all of 19 blend modes in total.
image
*Preview of the blend mode

3 The BlendModeV2 include normal, dissolve, darken, multiply, color burn, linear burn, darker color, lighten, screen, color dodge, linear dodge(add), lighter color, dodge, overlay, soft light, hard light, vivid light, linear light, pin light, hard mix, difference, exclusion, subtract, divide, hue, saturation, color, luminosity, grain extract, grain merge all of 30 blend modes in total.
Part of the code for BlendMode V2 is from Virtuoso Nodes for ComfyUI. Thanks to the original authors. image
*Preview of the Blend Mode V2

4 The RGB color described by hexadecimal RGB format, like '#FA3D86'.

5 The layer_image and layer_mask must be of the same size.

Stars

Star History Chart

statement

LayerStyle nodes follows the MIT license, Some of its functional code comes from other open-source projects. Thanks to the original author. If used for commercial purposes, please refer to the original project license to authorization agreement.

About

A set of nodes for ComfyUI that can composite layer and mask to achieve Photoshop like functionality.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published