Files
45uee-ComfyUI-Color_Transfer/__init__.py
T
Stefan Koehler 6dcfa314d6 adapt naming so that we are shown under
"Color Transfer" in the Node Library
2025-05-10 00:08:18 +02:00

17 lines
744 B
Python

from .color_transfer import PaletteTransferNode, PalleteTransferClustering, PaletteTransferReinhard, PaletteSoftTransfer, PaletteRbfTransfer, PaletteOptimalTransportTransfer, ReferenceTransferReinhard, ColorPaletteNode
NODE_CLASS_MAPPINGS = {
"PaletteTransfer": PaletteTransferNode,
"PalleteTransferClustering": PalleteTransferClustering,
"PaletteTransferReinhard": PaletteTransferReinhard,
"PalletteSoftTransfer": PaletteSoftTransfer,
"PaletteRbfTransfer": PaletteRbfTransfer,
"PaletteOptimalTransportTransfer": PaletteOptimalTransportTransfer,
"ColorTransferReinhard": ReferenceTransferReinhard,
"ColorPalette": ColorPaletteNode,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"PaletteTransfer": "Palette Transfer",
}