60 lines
2.3 KiB
Python
60 lines
2.3 KiB
Python
from .nodes import (
|
|
ApplyCircularConvPaddingModel,
|
|
ApplyCircularConvPaddingVAE,
|
|
C2EMaskedDiffNode,
|
|
C2ENode,
|
|
CreateSeamMask,
|
|
Crop360To180Node,
|
|
CropImageWithCoordsNode,
|
|
E2CNode,
|
|
E2ENode,
|
|
E2PNode,
|
|
MonoScopicToStereoNode,
|
|
Pad180To360Node,
|
|
PasteImageWithCoordsNode,
|
|
RollImageNode,
|
|
SplitFacesNode,
|
|
StackFacesNode,
|
|
StereoToMonoScopicNode,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"Cubemap to Equirectangular": C2ENode,
|
|
"Equirectangular to Cubemap": E2CNode,
|
|
"Equirectangular to Perspective": E2PNode,
|
|
"Equirectangular Rotation": E2ENode,
|
|
"Split Cubemap Faces": SplitFacesNode,
|
|
"Stack Cubemap Faces": StackFacesNode,
|
|
"Roll Image Axes": RollImageNode,
|
|
"Masked Diff C2E": C2EMaskedDiffNode,
|
|
"Crop Image with Coords": CropImageWithCoordsNode,
|
|
"Paste Image with Coords": PasteImageWithCoordsNode,
|
|
"Pad 180 to 360 Equirectangular": Pad180To360Node,
|
|
"Crop 360 to 180 Equirectangular": Crop360To180Node,
|
|
"Crop Stereo to Monoscopic": StereoToMonoScopicNode,
|
|
"Merge Monoscopic into Stereo": StereoToMonoScopicNode,
|
|
"Apply Circular Padding VAE": ApplyCircularConvPaddingVAE,
|
|
"Apply Circular Padding Model": ApplyCircularConvPaddingModel,
|
|
"Create Seam Mask": CreateSeamMask,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"Cubemap to Equirectangular": "Cubemap to Equirectangular",
|
|
"Equirectangular to Cubemap": "Equirectangular to Cubemap",
|
|
"Equirectangular to Perspective": "Equirectangular to Perspective",
|
|
"Equirectangular Rotation": "Equirectangular Rotation",
|
|
"Split Cubemap Faces": "Split Cubemap Faces",
|
|
"Stack Cubemap Faces": "Stack Cubemap Faces",
|
|
"Roll Image Axes": "Roll Image Axes",
|
|
"Masked Diff C2E": "Masked Diff C2E",
|
|
"Crop Image with Coords": "Crop Image with Coords",
|
|
"Paste Image with Coords": "Paste Image with Coords",
|
|
"Pad 180 to 360 Equirectangular": "Pad 180 to 360 Equirectangular",
|
|
"Crop 360 to 180 Equirectangular": "Crop 360 to 180 Equirectangular",
|
|
"Crop Stereo to Monoscopic": "Crop Stereo to Monoscopic",
|
|
"Merge Monoscopic into Stereo": "Merge Monoscopic into Stereo",
|
|
"Apply Circular Padding VAE": "Apply Circular Padding VAE",
|
|
"Apply Circular Padding Model": "Apply Circular Padding Model",
|
|
"Create Seam Mask": "Create Seam Mask",
|
|
}
|