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).
27 lines
1.1 KiB
Python
27 lines
1.1 KiB
Python
"""imgutils_nodes edge detection sub-package."""
|
|
from __future__ import annotations
|
|
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
try:
|
|
from .node_canny import ImgUtilsCanny
|
|
NODE_CLASS_MAPPINGS["ImgUtilsCanny"] = ImgUtilsCanny
|
|
NODE_DISPLAY_NAME_MAPPINGS["ImgUtilsCanny"] = "Imgutils Edge (Canny)"
|
|
except Exception:
|
|
import logging; logging.getLogger(__name__).warning("ImgUtilsCanny unavailable", exc_info=True)
|
|
|
|
try:
|
|
from .node_lineart import ImgUtilsLineart
|
|
NODE_CLASS_MAPPINGS["ImgUtilsLineart"] = ImgUtilsLineart
|
|
NODE_DISPLAY_NAME_MAPPINGS["ImgUtilsLineart"] = "Imgutils Edge (Lineart)"
|
|
except Exception:
|
|
import logging; logging.getLogger(__name__).warning("ImgUtilsLineart unavailable", exc_info=True)
|
|
|
|
try:
|
|
from .node_lineart_anime import ImgUtilsLineartAnime
|
|
NODE_CLASS_MAPPINGS["ImgUtilsLineartAnime"] = ImgUtilsLineartAnime
|
|
NODE_DISPLAY_NAME_MAPPINGS["ImgUtilsLineartAnime"] = "Imgutils Edge (Lineart Anime)"
|
|
except Exception:
|
|
import logging; logging.getLogger(__name__).warning("ImgUtilsLineartAnime unavailable", exc_info=True)
|