add background restorer

This commit is contained in:
KBRASK
2025-08-17 13:47:46 +08:00
parent feb48d12c6
commit 05bfe6bb17
2 changed files with 135 additions and 0 deletions
+3
View File
@@ -17,15 +17,18 @@ NODE_DISPLAY_NAME_MAPPINGS = {}
try:
from .nodes import FaceProcessorLoader, ApplyFaceProcessor
from .videoinput import VideoInputPreprocessor
from .videocombine import VideoBackgroundRestorer
# Map the class names to the classes themselves
NODE_CLASS_MAPPINGS["FaceProcessorLoader"] = FaceProcessorLoader
NODE_CLASS_MAPPINGS["ApplyFaceProcessor"] = ApplyFaceProcessor
NODE_CLASS_MAPPINGS["VideoInputPreprocessor"] = VideoInputPreprocessor
NODE_CLASS_MAPPINGS["VideoBackgroundRestorer"] = VideoBackgroundRestorer
NODE_DISPLAY_NAME_MAPPINGS["FaceProcessorLoader"] = "Stand-In Processor Loader"
NODE_DISPLAY_NAME_MAPPINGS["ApplyFaceProcessor"] = "Apply Stand-In Processor"
NODE_DISPLAY_NAME_MAPPINGS["VideoInputPreprocessor"] = "Stand-In VideoInputPreprocessor"
NODE_DISPLAY_NAME_MAPPINGS["VideoBackgroundRestorer"] = "Stand-In Background Restorer"
logger.info("Successfully loaded all Stand-In nodes.")