Files

33 lines
1.3 KiB
Python

from .nodes.grayscale_image import GrayscaleImage
from .nodes.brightness_transparency import BrightnessTransparency
from .nodes.binarize_image import BinarizeImage
from .nodes.binarize_image_using_otsu import BinarizeImageUsingOtsu
from .nodes.antialiasing_image import AntialiasingImage
from .nodes.remove_white_background_noise import RemoveWhiteBackgroundNoise
from .nodes.alpha_to_grayscale import AlphaToGrayscale
from .nodes.alpha_flatten import AlphaFlatten
NODE_CLASS_MAPPINGS = {
"BrightnessTransparency": BrightnessTransparency,
"BinarizeImage": BinarizeImage,
"BinarizeImageUsingOtsu": BinarizeImageUsingOtsu,
"GrayscaleImage": GrayscaleImage,
"AntialiasingImage": AntialiasingImage,
"RemoveWhiteBackgroundNoise": RemoveWhiteBackgroundNoise,
"AlphaToGrayscale": AlphaToGrayscale,
"AlphaFlatten": AlphaFlatten,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"BrightnessTransparency": "BrightnessTransparency",
"BinarizeImage": "BinarizeImage",
"BinarizeImageUsingOtsu": "BinarizeImageUsingOtsu",
"GrayscaleImage": "GrayscaleImage",
"AntialiasingImage": "AntialiasingImage",
"RemoveWhiteBackgroundNoise": "RemoveWhiteBackgroundNoise",
"AlphaToGrayscale": "AlphaToGrayscale",
"AlphaFlatten": "AlphaFlatten",
}
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]