add background restorer
This commit is contained in:
@@ -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.")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user