34 lines
1.3 KiB
Python
34 lines
1.3 KiB
Python
from .nodes.crop_and_resize_nodes import (
|
|
SmartImageResize,
|
|
ResizeToMultiple,
|
|
ExtendCanvasByPercentage,
|
|
ImageCropByPercentage,
|
|
RemovePadding,
|
|
)
|
|
from .nodes.mask_nodes import MaskCropByPercentage, ApplyMaskToImage
|
|
from .nodes.extractor_nodes import ImageResolutionExtractor, ImageAspectRatioExtractor
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"ImageCropByPercentage": ImageCropByPercentage,
|
|
"MaskCropByPercentage": MaskCropByPercentage,
|
|
"ExtendCanvasByPercentage": ExtendCanvasByPercentage,
|
|
"ResizeToMultiple": ResizeToMultiple,
|
|
"ImageResolutionExtractor": ImageResolutionExtractor,
|
|
"ImageAspectRatioExtractor": ImageAspectRatioExtractor,
|
|
"SmartImageResize": SmartImageResize,
|
|
"RemovePadding": RemovePadding,
|
|
"ApplyMaskToImage": ApplyMaskToImage,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"ImageCropByPercentage": "Image Crop By Percentage",
|
|
"MaskCropByPercentage": "Mask Crop By Percentage",
|
|
"ExtendCanvasByPercentage": "Extend Canvas By Percentage",
|
|
"ResizeToMultiple": "Resize To Multiple",
|
|
"ImageResolutionExtractor": "Image Resolution Extractor",
|
|
"ImageAspectRatioExtractor": "Image Aspect Ratio Extractor",
|
|
"SmartImageResize": "Smart Image Resize",
|
|
"RemovePadding": "Remove Padding",
|
|
"ApplyMaskToImage": "Apply Mask To Image",
|
|
}
|