Extensions/castcut-nodes
ComfyUI Extension

castcut-nodes

ComfyUI nodes for Castcut: in-job pose scoring, best-of-two pick, face distance and cut-out mask repair.

By doodersrage·Created about 24 hours ago·Updated about 13 hours ago· 1
doodersrage/castcut-nodes
Nodes5
On cloudLocal install
CategoryCastcut
Stars1
Updatedabout 13 hours ago

Nodes (5)

Castcut Face Distance

Is this still the same face?

Castcut
Castcut Mask Repair

Filling the holes your matte left behind

Castcut
Castcut Pick Best

Keep the take that actually hit the pose

Castcut
Castcut Pose Score

How close is this take to your guide pose?

Castcut
Castcut Report

The node whose real output is a history entry

Castcut
Readme

Castcut nodes for ComfyUI

Optional. Castcut runs a few checks after a still lands — the pose check (DWPose + limb angles + posture), Best of two for hard poses, the cut-out's matte repair. Without this pack each is a separate ComfyUI call (and Best of two is two jobs, queued one after the other). With it they run inside the job:

  • Best of two in one job (Day → Best of two for hard poses): the sampler renders both takes as a batch (one model load, one encode), DWPose reads both, CastcutPoseScore scores them against the guide, CastcutPickBest hands the closer take to the job's SaveImage and CastcutReport saves the other take and writes both scores into the job's history, where the app picks them up.
  • Cut-out in one job (Isolate on white / plates): BiRefNet matte → CastcutMaskRepair (hole fill from the photo's colours) → the original pixels composited onto the fill — no mask round trip.

The app checks ComfyUI's object_info; when the nodes are missing it queues exactly the graphs it always did.

Nodes

| Node | In | Out | | --- | --- | --- | | CastcutPoseScore | POSE_KEYPOINT (DWPose), guide_json | per-image results JSON (the app's PoseMatchResult), first score | | CastcutPickBest | IMAGE batch, scores JSON | best image, other image, best index, report JSON | | CastcutFaceDistance | ANALYSIS_MODELS (ComfyUI_FaceAnalysis), reference, images | cosine distance per image (100 = no face) | | CastcutMaskRepair | image, mask, fill #rrggbb, regrow edges, optional LoadImage mask | composite, repaired mask, report JSON | | CastcutReport | report JSON, optional alternate image | UI output castcut (read from /history), alternate saved to output |

CastcutPoseScore and CastcutMaskRepair are ports of src/lib/pose-score.ts, pose-limb-score.ts, pose-posture.ts and src/lib/isolate-mask.ts. They stay in step through shared test vectors (tests/vectors/*.json) that both test suites read.

Needs only what ComfyUI already has (numpy, PIL; cv2 is used when present). Best of two needs comfyui_controlnet_aux (DWPose); CastcutFaceDistance needs ComfyUI_FaceAnalysis with insightface and fails with a clear message without it.

Install

The pack is one file, castcut_nodes.py (plus this folder's __init__.py when installed as a folder). Pick one way and restart ComfyUI afterwards.

From the Castcut app (easiest): Settings → ComfyUI → Castcut nodes shows whether ComfyUI has the pack and which version, installs it with ComfyUI-Manager when it can, restarts ComfyUI (after asking, and only when its queue is empty), or hands you a copy-paste command for your setup. See docs/castcut-nodes.md.

  1. ComfyUI-Manager by name (from the Comfy Registry): Manager → Custom Nodes Manager → search Castcut nodes → Install; or comfy node install castcut-nodes.

  2. ComfyUI-Manager → Install via Git URL: https://github.com/doodersrage/castcut-nodes. Manager 3.41+ allows Git-URL installs only with allow_git_url_install = true under [default] in ComfyUI/user/__manager/config.ini and ComfyUI listening on 127.0.0.1. The repository holds just this pack.

  3. Copy the file (smallest; works when the ComfyUI service user can't read your home folder):

    sudo install -m 644 comfyui-nodes/castcut/castcut_nodes.py \
  /opt/comfyui/custom_nodes/castcut_nodes.py
sudo systemctl restart comfyui

    (Use your ComfyUI path; a desktop ComfyUI: copy into ComfyUI/custom_nodes/; Windows portable: ComfyUI_windows_portable\ComfyUI\custom_nodes\. A running Castcut serves the file at /api/castcut-nodes/file.)

  4. Symlink the folder (ComfyUI must be able to read the repo):

    ln -s "$PWD/comfyui-nodes/castcut" /path/to/ComfyUI/custom_nodes/castcut

Check http://127.0.0.1:8188/object_info/CastcutPoseScore returns the node; its description ends with [castcut-nodes 1.1.0], which is how the app reads the installed version. To update, copy / pull / update again and restart. Install it one way only — a castcut_nodes.py file and a castcut folder side by side both load.

Without the pack the app checks object_info, finds the nodes missing and queues exactly the graphs it always did: Best of two as two jobs, cut-outs repaired in the app.

Versions and publishing

CASTCUT_VERSION in castcut_nodes.py, version in pyproject.toml and the app's CASTCUT_NODES_BUNDLED_VERSION (src/lib/castcut-nodes-setup.ts) move together; the tests check it. pyproject.toml carries the Comfy Registry metadata ([tool.comfy], PublisherId is a placeholder until a publisher exists). scripts/castcut-nodes-release.sh <dir> copies this folder, the licence and a standalone README note into <dir>, ready to push as its own repository and publish with comfy node publish.

Tests

Plain Python, no ComfyUI or torch:

python3 -m pytest comfyui-nodes/castcut/tests      # or
python3 -m unittest discover -s comfyui-nodes/castcut/tests

The vectors come from the TypeScript code. After changing the pose check or the mask repair:

node --import tsx scripts/castcut-test-vectors.mts   # keeps the real rows, redoes the synthetic ones
npm test                                             # src/lib/castcut-vectors.test.ts
python3 -m pytest comfyui-nodes/castcut/tests        # the port follows?

--dataset <pose-calibration.json> --every 6 and --real-mask <still.png>:<matte.png> refresh the real rows (see the script header).