Spaces:
Runtime error
Runtime error
File size: 3,258 Bytes
f2dbf59 |
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 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 |
import sys
from pathlib import Path
from .utils import here
import platform
sys.path.insert(0, str(Path(here, "src").resolve()))
from .nodes.image.load_node import LoadImageNode
from .nodes.image.save_node import SaveImageNode
from .nodes.image.save_node import SaveImagesNode
from .nodes.image.size_node import ImageResizeNode
from .nodes.image.size_node import ImageSizeNode
from .nodes.image.size_node import ConstrainImageNode
from .nodes.image.watermark_node import WatermarkNode
from .nodes.mask.mask_node import MaskAddNode
from .nodes.video.load_node import LoadVideoNode
from .nodes.video.save_node import SaveVideoNode
from .nodes.video.info_node import VideoInfoNode
from .nodes.video.batch_node import BatchManagerNode
from .nodes.preprocessor.canny_node import Canny_Preprocessor
from .nodes.preprocessor.lineart_node import LineArt_Preprocessor
from .nodes.preprocessor.lineart_node import Lineart_Standard_Preprocessor
from .nodes.preprocessor.midas_node import MIDAS_Depth_Map_Preprocessor
from .nodes.preprocessor.dwpose_node import DWPose_Preprocessor, AnimalPose_Preprocessor
from .nodes.preprocessor.densepose_node import DensePose_Preprocessor
from .nodes.face.face_enhance_node import GFPGANNode
from .nodes.other.vram_node import PurgeVRAMNode
NODE_CLASS_MAPPINGS = {
"PurgeVRAMNode": PurgeVRAMNode,
"GFPGANNode": GFPGANNode,
"MaskAddNode": MaskAddNode,
"ImageLoader": LoadImageNode,
"ImageSaver": SaveImageNode,
"ImagesSaver": SaveImagesNode,
"ImageResize": ImageResizeNode,
"ImageSize": ImageSizeNode,
"WatermarkNode": WatermarkNode,
"VideoLoader": LoadVideoNode,
"VideoSaver": SaveVideoNode,
"VideoInfo": VideoInfoNode,
"BatchManager": BatchManagerNode,
"ConstrainImageNode": ConstrainImageNode,
"DensePosePreprocessor": DensePose_Preprocessor,
"DWPosePreprocessor": DWPose_Preprocessor,
"AnimalPosePreprocessor": AnimalPose_Preprocessor,
"MiDaSDepthPreprocessor": MIDAS_Depth_Map_Preprocessor,
"CannyPreprocessor": Canny_Preprocessor,
"LineArtPreprocessor": LineArt_Preprocessor,
"LineartStandardPreprocessor": Lineart_Standard_Preprocessor,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"PurgeVRAMNode":"PurgeVRAMNode",
"GFPGANNode": "GFPGANNode",
"MaskAddNode": "MaskAddNode",
"ImageLoader": "Image Load",
"ImageSaver": "Image Save",
"ImagesSaver": "Image List Save",
"ImageResize": "Image Resize",
"ImageSize": "Image Size",
"WatermarkNode": "Watermark",
"VideoLoader": "Video Load",
"VideoSaver": "Video Save",
"VideoInfo": "Video Info",
"BatchManager": "Batch Manager",
"ConstrainImageNode": "Image Constrain",
"DensePosePreprocessor": "DensePose Estimator",
"DWPosePreprocessor": "DWPose Estimator",
"AnimalPosePreprocessor": "AnimalPose Estimator",
"MiDaSDepthPreprocessor": "MiDaS Depth Estimator",
"CannyPreprocessor": "Canny Edge Estimator",
"LineArtPreprocessor": "Realistic Lineart",
"LineartStandardPreprocessor": "Standard Lineart",
}
if platform.system() == "Darwin":
WEB_DIRECTORY = "./web"
__all__ = ["WEB_DIRECTORY", "NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
else:
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"] |