ComfyUI Node
TS Color Match
Transfer the colour palette of a reference image onto a target batch.
TS Color Match
- reference
- target
- image
◄modemkl►
◄deviceauto►
◄strength1.0►
◄enabletrue►
◄match_masknone►
◄mask_size32►
◄compute_max_side1024►
◄mkl_sample_points300000►
◄sinkhorn_max_points2048►
◄reuse_referencetrue►
◄temporal_smoothingtrue►
◄chunk_size4►
◄loggingfalse►
CategoryTS/Image/Color
Inputs (15)
| Name | Type | Default | Description |
|---|---|---|---|
| reference | IMAGE | Reference whose colors the target is matched to. With batch=1 the statistics can be reused across a whole video. | |
| target | IMAGE | Frames/images that get recolored toward the reference. | |
| mode | COMBO | mkl | mkl is faster and more stable; sinkhorn is more accurate but heavier on memory and time. |
| device | COMBO | auto | auto = GPU if available, otherwise CPU. In GPU mode large tensors stay on CPU to guard against OOM. |
| strength | FLOAT | 1.00–1 | Softly blends the original and the color-corrected result. |
| enable | BOOLEAN | true | When off, the node returns the input target unchanged. |
| match_mask | COMBO | none | rectangle/ellipse sample only the image border of width mask_size, for stabilization. |
| mask_size | INT | 320–512 | Used only when match_mask is rectangle or ellipse. |
| compute_max_side | INT | 10240–4096 | Compute A,b on a downscaled copy (e.g. 512–1024). Saves memory at 4K with almost no quality loss. |
| mkl_sample_points | INT | 3000000–2000000 | 200k–500k recommended. At 0 all pixels are used, which can OOM at 4K. |
| sinkhorn_max_points | INT | 20480–65536 | Usually 1024–2048 for 4K. Too large a value can OOM. |
| reuse_reference | BOOLEAN | true | Speeds up and stabilizes video when one reference is shared across the whole batch. |
| temporal_smoothing | BOOLEAN | true | EMA smoothing of the color transforms between frames. Enable for video (suppresses flicker); disable for a batch of UNrelated images, otherwise per-frame corrections bleed into each other. |
| chunk_size | INT | 40–256 | 4–8 recommended for long 4K videos. 0 processes the whole batch at once. |
| logging | BOOLEAN | false | When on, logs processing stages, chunk index and GPU memory state to the console. |
Outputs (1)
| Name | Type | Description |
|---|---|---|
| image | IMAGE | — |