ComfyUI Extension: ComfyUI-WarperNodes
NODES: DWPose Scaler (Warper), Mouth Mask from Pose (Warper), Facial Part Mask from Pose (Warper)
Custom Nodes (0)
README
ComfyUI Warper Nodes
A collection of general-purpose nodes for ComfyUI.
Current Nodes
-
DWPose Scaler (Warper): Provides advanced scaling and adjustment options for DWPose keypoints. (Category:
Warper Tools
) -
Flow Visualizer (Warper): Calculates optical flow between image sequences using the RAFT model and provides visual representations (color-coded flow images) for both forward and backward flow. Ideal for understanding motion in your generations. (Category:
Warper Tools/Flow
)- Model Requirement: This node requires a RAFT model checkpoint.
-
Smart Video Batcher (Warper): Splits image sequences into smaller batches with special padding for the final batch. (Category:
Warper Tools/Looping
) -
Get Batch By Index (Warper): Selects a single image batch from a list of batches by its index. (Category:
Warper Tools/Looping
) -
Smart Overlapping Batcher (Warper): Creates overlapping batches from an image sequence to help maintain temporal consistency. (Category:
Warper Tools/Looping
) -
Mouth Mask from Pose (Warper): Generates a circular mask around the mouth area based on facial keypoints. (Category:
Warper Tools/DWMasking
) -
Facial Part Mask from Pose (Warper): Creates masks for various facial parts (entire face, mouth, eyes) using different shapes (convex hull, ellipse, etc.) based on facial keypoints. (Category:
Warper Tools/DWMasking
)
Installation
-
Navigate to your ComfyUI
custom_nodes
directory. -
Clone this repository:
git clone https://github.com/AIWarper/ComfyUI-WarperNodes
-
Install RAFT Model (Required for Flow Visualizer Node):
- Download the RAFT model checkpoint:
Tartan-C-T-TSKH-spring540x960-M.pth
- Download Link: Google Drive Folder (Look for the specified
.pth
file within this folder.)
- Download Link: Google Drive Folder (Look for the specified
- Create a
raft
folder inside your ComfyUImodels
directory if it doesn't already exist:ComfyUI/models/raft/
- Place the downloaded
Tartan-C-T-TSKH-spring540x960-M.pth
file into thisComfyUI/models/raft/
directory.
- Download the RAFT model checkpoint:
-
Install any Python package dependencies (if listed in
requirements.txt
):cd ComfyUI-WarperNodes pip install -r requirements.txt
(If you use ComfyUI's portable version, you might need to use its embedded Python, e.g.,
path/to/ComfyUI/python_embeded/python.exe -m pip install -r requirements.txt
. If arequirements.txt
file is not present, this step may not be needed.) -
Restart ComfyUI.
Usage Notes
- Ensure the RAFT model is correctly placed as described above for the Flow Visualizer (Warper) node to function.
- The nodes will appear under the "Warper Tools" category in the ComfyUI "Add Node" menu.
TODO:
- Add example images/gifs for other nodes.
- More detailed usage instructions for each node.