File size: 1,414 Bytes
1e3b872 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
from .nodes import *
from .startup_utils import symlink_web_dir
# A dictionary that contains all nodes you want to export with their names
# NOTE: names should be globally unique
NODE_CLASS_MAPPINGS = {
"FS: Fit Size From Int": FitSize,
"FS: Fit Size From Image": FitSizeFromImage,
"FS: Fit Image And Resize": FitResizeLatent,
"FS: Load Image And Resize To Fit": LoadToFitResizeLatent,
"FS: Pick Image From Batch": RandomImageFromBatch,
"FS: Pick Image From Batches": RandomImageFromBatches,
"FS: Pick Image From List": RandomImageFromList,
"FS: Crop Image Into Even Pieces": CropImageIntoEvenPieces,
"FS: Image Region To Mask": ImageRegionMask,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"FS: Fit Size From Int": "Fit Size From Int (FS)",
"FS: Fit Size From Image": "Fit Size From Image (FS)",
"FS: Fit Image And Resize": "Fit Image And Resize (FS)",
"FS: Load Image And Resize To Fit": "Load Image And Resize To Fit (FS)",
"FS: Pick Image From Batch": "Pick Image From Batch (FS)",
"FS: Pick Image From Batches": "Pick Image From Batches (FS)",
"FS: Pick Image From List": "Pick Image From List (FS)",
"FS: Crop Image Into Even Pieces": "Crop Image Into Even Pieces (FS)",
"FS: Image Region To Mask": "Image Region To Mask (FS)",
}
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
EXTENSION_NAME = "Fitsize"
WEB_DIRECTORY = "./js"
|