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).
20 lines
728 B
Python
20 lines
728 B
Python
"""imgutils_nodes compare sub-package."""
|
|
from __future__ import annotations
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
try:
|
|
from .node_ccip import ImgUtilsCCIP
|
|
NODE_CLASS_MAPPINGS["ImgUtilsCCIP"] = ImgUtilsCCIP
|
|
NODE_DISPLAY_NAME_MAPPINGS["ImgUtilsCCIP"] = "Imgutils Compare (CCIP)"
|
|
except Exception:
|
|
import logging; logging.getLogger(__name__).warning("ImgUtilsCCIP unavailable", exc_info=True)
|
|
|
|
try:
|
|
from .node_lpips import ImgUtilsLPIPS
|
|
NODE_CLASS_MAPPINGS["ImgUtilsLPIPS"] = ImgUtilsLPIPS
|
|
NODE_DISPLAY_NAME_MAPPINGS["ImgUtilsLPIPS"] = "Imgutils Compare (LPIPS)"
|
|
except Exception:
|
|
import logging; logging.getLogger(__name__).warning("ImgUtilsLPIPS unavailable", exc_info=True)
|