26 lines
1.2 KiB
Python
26 lines
1.2 KiB
Python
from .color_transfer import PaletteTransferNode, PalleteTransferClustering, PaletteTransferReinhard, PaletteSoftTransfer, PaletteRbfTransfer, PaletteOptimalTransportTransfer, ReferenceTransferReinhard, ColorPaletteNode, ExtractPaletteNode
|
|
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"PaletteTransfer": PaletteTransferNode,
|
|
"PalleteTransferClustering": PalleteTransferClustering,
|
|
"PaletteTransferReinhard": PaletteTransferReinhard,
|
|
"PalletteSoftTransfer": PaletteSoftTransfer,
|
|
"PaletteRbfTransfer": PaletteRbfTransfer,
|
|
"PaletteOptimalTransportTransfer": PaletteOptimalTransportTransfer,
|
|
"ColorTransferReinhard": ReferenceTransferReinhard,
|
|
"ColorPalette": ColorPaletteNode,
|
|
"ExtractPalette": ExtractPaletteNode
|
|
}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"PaletteTransfer": "Palette Transfer",
|
|
"PalleteTransferClustering": "Palette Transfer Clustering",
|
|
"PaletteTransferReinhard": "Palette Transfer Reinhard",
|
|
"PalletteSoftTransfer": "Palette Soft Transfer",
|
|
"PaletteRbfTransfer": "Palette RBF Transfer",
|
|
"PaletteOptimalTransportTransfer": "Palette Optimal Transport Transfer",
|
|
"ColorTransferReinhard": "Reference Transfer Reinhard",
|
|
"ColorPalette": "Color Palette",
|
|
"ExtractPalette": "Extract Palette",
|
|
}
|