Nodes Browser

ComfyDeploy: How ComfyUI-Nudenet works in ComfyUI?

What is ComfyUI-Nudenet?

Nodes for NSFW content filtering

How to install it in ComfyDeploy?

Head over to the machine page

  1. Click on the "Create a new machine" button
  2. Select the Edit build steps
  3. Add a new step -> Custom Node
  4. Search for ComfyUI-Nudenet and select it
  5. Close the build step dialig and then click on the "Save" button to rebuild the machine

Hits

ComfyUI-Nudenet

Nodes for NSFW content filtering.

Implementation is base on notAI-tech/NudeNet and vladmandic/sd-extension-nudenet

Features

  • Simple integration with ComfyUI workflow
  • Exposure filtering with customizable thresholds

Installation

  1. Clone this repository into your ComfyUI custom nodes directory:
cd ComfyUI/custom_nodes/
git clone https://github.com/phuvinh010701/ComfyUI-NudeNet
  1. Install required dependencies
pip install -r requirements.txt
  1. Create folder, download models Download model from here and put into models/Nudenet/

Available Nodes

NudenetModelLoader

Loads and initializes the NudeNet ONNX model.

  • Inputs:

    model: Select from available NudeNet models in the models directory

  • Outputs:

    NUDENET_MODEL: Initialized model for use in other nodes

FilteredLabel

Configure which labels to filter in the detection process.

  • Inputs:

    Multiple boolean toggles for each available label in LABELS_CLASSIDS_MAPPING Each label can be enabled (True) or disabled (False)

  • Outputs:

    FILTERED_LABELS: List of enabled label class IDs for filtering

ApplyNudenet

Applies NSFW content filtering to images.

  • Required Inputs:

    nudenet_model (NUDENET_MODEL): Loaded NudeNet model
    image (IMAGE): Input image to process
    censor_method: Method for censoring detected areas
    filtered_labels (FILTERED_LABELS): Labels to filter from FilteredLabel node
    min_score (FLOAT): Detection confidence threshold (0.0 - 1.0, default: 0.2)
    blocks (INT): Number of blocks for pixelation (1 - 100, default: 3)

  • Optional Inputs:

    overlay_image (IMAGE): Image to use as overlay for censoring
    overlay_strength (FLOAT): Strength of the overlay effect (0.0 - 10.0, default: 1.0)

  • Outputs:

    IMAGE: Processed image with applied filtering

Usage Example

Using example workflow from ./example/nudenet_base_example.json Or drag and drop this image into ComfyUI

<img src="./example/workflow.png" alt="example workflow" width="300"/>

Contributing

Contributions are welcome! Please feel free to submit a Pull Request.