xiaden
7d82aa22e6
Sub-packages: - edge/ — Canny, Lineart, LineartAnime (3 nodes) - restore/ — SCUNet, NAFNet, Adversarial (3 nodes) - compare/ — CCIP, LPIPS (2 nodes) - validate/ — Classify (11 ops), Check (5 ops), Metric (2 ops) - tagging/ — WD14, DeepDanbooru, DeepGelbooru, MLDanbooru, Camie, PixAI (6 nodes) Root nodes: - ImgUtilsOCR — PaddleOCR with heat/box threshold - ImgUtilsDetect — 10 detection types + confidence knob - ImgUtilsPose — DWPose + auto_detect toggle - ImgUtilsSegment — isnetis + scale knob - ImgUtilsTags — tag string processing - ImgUtilsCDC — upscaling + tile_size - ImgUtilsCensor — NSFW censoring with method/confidence/body-part toggles - ImgUtilsAlign — resize to max dimension - ImgUtilsSqueeze — transparency-based auto-crop Every node exposes its native function parameters as knobs. Removed mega-nodes: transform (11 ops), compare (3 ops), validate (18 ops).
34 lines
1.7 KiB
Python
34 lines
1.7 KiB
Python
"""Node: ImgUtilsCensor — NSFW content censoring."""
|
|
from __future__ import annotations
|
|
from comfy_api.latest import io
|
|
from ..utils import comfy_to_pil, pil_to_comfy
|
|
|
|
class ImgUtilsCensor(io.ComfyNode):
|
|
METHODS = ["pixelate", "blur", "color"]
|
|
|
|
@classmethod
|
|
def define_schema(cls) -> io.Schema:
|
|
return io.Schema(
|
|
node_id="ImgUtilsCensor", display_name="Imgutils Censor",
|
|
category="imgutils/transform",
|
|
description="Auto-detect and censor NSFW content. Applies pixelation, blur, or solid color.",
|
|
search_aliases=["censor", "nsfw", "pixelate", "blur", "mosaic"],
|
|
inputs=[
|
|
io.Image.Input("image", tooltip="Input image"),
|
|
io.Combo.Input("method", options=cls.METHODS, default="pixelate", tooltip="Censoring method."),
|
|
io.Boolean.Input("nipple_f", default=True, tooltip="Censor female nipples."),
|
|
io.Boolean.Input("penis", default=True, tooltip="Censor penises."),
|
|
io.Boolean.Input("pussy", default=True, tooltip="Censor vaginas."),
|
|
io.Float.Input("conf_threshold", default=0.3, min=0.0, max=1.0, step=0.05, tooltip="Detection confidence threshold."),
|
|
],
|
|
outputs=[io.Image.Output(display_name="image")],
|
|
)
|
|
|
|
@classmethod
|
|
def execute(cls, image, method, nipple_f, penis, pussy, conf_threshold) -> io.NodeOutput:
|
|
from imgutils.operate import censor_nsfw
|
|
pil = comfy_to_pil(image.numpy() if hasattr(image, "numpy") else image)
|
|
result = censor_nsfw(pil, method=str(method), nipple_f=bool(nipple_f), penis=bool(penis),
|
|
pussy=bool(pussy), conf_threshold=float(conf_threshold))
|
|
return io.NodeOutput(pil_to_comfy(result))
|