ryanontheinside
33d26c4808
this commit includes -a stream-optimized node for detecting facial landmarks -node for drawing the mesh on an image from the landmark data -node for creating masks from the landmark data
19 lines
706 B
Python
19 lines
706 B
Python
"""ComfyUI Stream Pack Nodes."""
|
|
|
|
from .node_wrappers import feature_bank
|
|
from .node_wrappers import facemesh
|
|
# Collect all NODE_CLASS_MAPPINGS and NODE_DISPLAY_NAME_MAPPINGS from submodules.
|
|
NODE_CLASS_MAPPINGS = {}
|
|
NODE_DISPLAY_NAME_MAPPINGS = {}
|
|
|
|
# Import and update mappings from submodules.
|
|
for module in [feature_bank, facemesh]:
|
|
if hasattr(module, "NODE_CLASS_MAPPINGS"):
|
|
NODE_CLASS_MAPPINGS.update(module.NODE_CLASS_MAPPINGS)
|
|
if hasattr(module, "NODE_DISPLAY_NAME_MAPPINGS"):
|
|
NODE_DISPLAY_NAME_MAPPINGS.update(module.NODE_DISPLAY_NAME_MAPPINGS)
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS["StreamPack"] = "Stream Pack Nodes"
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|