Extensions/ComfyUI Temporal Mask Tools
ComfyUI Extension

ComfyUI Temporal Mask Tools

Utility collection of ComfyUI V3 nodes for stabilizing temporal segmentation masks while staying deterministic and torch-only.

By nomadoor·Created 10 months ago·Updated 10 months ago· 2
nomadoor/ComfyUI-Temporal-Mask-Tools
Nodes2
On cloudLocal install
CategoryTemporalMask/Operations
Stars2
Updated10 months ago

Nodes (2)

Temporal Mask Remove Short Objects
TemporalMask/Operations
Temporal Mask Union
TemporalMask/Operations
Readme

ComfyUI Temporal Mask Tools

Temporal-mask cleaning tools for ComfyUI: union noisy frames to smooth flicker, then drop one-off specks so your mask sequences stay stable.

Sample Videos

| Original Video | Detection (Florence-2) | | --- | --- | | sample_video3 | mask_sample |

| After Union | After Remove Short Objects | | --- | --- | | mask_union | mask_remove |

Nodes

Temporal Mask Union (TemporalMaskUnion)

Combines nearby frames in a temporal mask sequence to suppress flicker.

| Input | Type | Default | Notes | | --- | --- | --- | --- | | mask_batch | MASK | required | Accepts (frames, H, W), (batch, frames, H, W), or (H, W) tensors. | | radius | INT | 2 | Temporal half-window size. | | mode | STRING | "or" | "or" keeps any active frame, "majority" uses the threshold. | | threshold | INT | 3 | Minimum active frames within the window when mode="majority". |

Temporal Mask Remove Short Objects (TemporalMaskRemoveShortObjects)

Drops one-frame flicker or tiny specks by combining per-frame connected-component filtering with temporal run-length pruning.

| Input | Type | Default | Notes | | --- | --- | --- | --- | | mask_batch | MASK | required | Supports (H, W), (frames, H, W), (batch, frames, H, W) tensors. | | min_duration | INT | 2 | Minimum consecutive frames required to keep a pixel active. | | min_area_pixels | INT | 10 | Connected components smaller than this pixel count are removed. |

Usage

  1. Clone into ComfyUI/custom_nodes and restart ComfyUI.
  2. Drop the desired node into your graph and connect it to a mask sequence batch.
  3. Optional: enable node-level debug_output (when available) while tuning parameters.