nodes/facok-ComfyUI-LCS

Fork 0

Files

T

facok 98bd528bb4 Add README in English and Chinese

2026-03-17 05:44:36 +08:00

6.0 KiB

Raw Blame History

ComfyUI-LCS

基于潜在颜色子空间（Latent Color Subspace）的 FLUX 无训练颜色控制。

基于论文 "The Latent Color Subspace"（ICML 2026）。论文发现 FLUX 64 维潜在 patch 空间中的颜色信息完全存在于一个 3 维子空间（PCA 捕获 100% 颜色方差），剩余 61 维编码结构和细节，与颜色正交。

本插件在扩散采样过程中直接操作 3D LCS 来控制颜色 —— 无需训练、无需 LoRA、无需后处理。

English README

功能

颜色引导 — 将生成图像的颜色推向任意目标颜色
批量多色 — 对批次中的每张图像施加不同颜色
色调调整 — 对比度、亮度、饱和度、色温，支持一键预设
局部控制 — 可选遮罩输入，实现区域性颜色控制
潜在颜色预览 — 无需 VAE 解码即可可视化颜色结构
步骤观察器 — 保存每步颜色预览图，用于观察扩散过程

安装

克隆到 ComfyUI 自定义节点目录：

cd ComfyUI/custom_nodes
git clone https://github.com/YOUR_USERNAME/ComfyUI-LCS.git

安装依赖（通常 ComfyUI 中已自带）：

pip install einops safetensors

节点一览

校准

节点	说明
LCS Calibrate	通过 PCA 从 FLUX VAE 计算 LCS 基底和锚点颜色。自动保存至 `data/lcs_calibration.safetensors`。
LCS Load Data	加载已缓存的校准数据，或在提供 VAE 时自动校准。

只需校准一次，结果会缓存并在后续会话中复用。

干预

节点	说明
LCS Color Intervene	在生成过程中将颜色引导至目标色。支持 Type I（LCS 平移）、Type II（HSL 偏移）或插值模式。
LCS Color Batch	对每个批次项施加不同目标颜色。额外输出 `batch_size` 可连接 EmptyLatentImage。
LCS Tone Adjust	调整对比度、亮度、饱和度和色温。带预设下拉菜单，滑条实时同步。

观察

节点	说明
LCS Preview Colors	将潜在颜色解码为 RGB 预览图，无需 VAE 解码。
LCS Step Observer	将每步颜色预览 PNG 保存至 ComfyUI 临时目录，用于调试。

色调预设

从下拉菜单选择预设后，滑条会实时更新数值。选择预设后仍可微调滑条。选择 Custom 可完全手动设置，不受预设影响。

预设	对比度	亮度	饱和度	色温
Base	1.0	0.0	1.0	0.0
Cinematic	1.20	-0.05	0.90	0.05
HDR	1.40	0.0	1.20	0.0
Vivid	1.10	0.0	1.50	0.0
Dramatic	1.50	-0.10	0.85	0.0
Low Key	1.30	-0.20	0.80	0.0
High Key	0.80	0.20	0.90	0.0
Warm	1.0	0.0	1.0	0.15
Cool	1.0	0.0	1.0	-0.15
Desaturated	1.0	0.0	0.40	0.0

快速开始

基本颜色控制

LCS Load Data → LCS Color Intervene → KSampler
                       ↑
                  （选择颜色）

添加 LCS Load Data — 连接 FLUX VAE（首次运行会自动校准）
添加 LCS Color Intervene — 连接 MODEL 和 LCS_DATA
选择目标颜色，设置强度（默认 1.0）
将输出 MODEL 连接到 KSampler

色调调整

LCS Load Data → LCS Tone Adjust → KSampler
                      ↑
              （选择预设或调整滑条）

添加 LCS Load Data → LCS Tone Adjust
选择预设（如 "Cinematic"）或使用 Custom 模式
按需微调滑条

批量多色生成

LCS Load Data → LCS Color Batch → KSampler
                      ↓
                  batch_size → EmptyLatentImage

输入逗号分隔的十六进制颜色（如 #FF0000,#00FF00,#0000FF），每个颜色对应一个批次项。

干预模式

模式	说明	适用场景
interpolated（默认）	以 sigma 为权重混合 Type I 和 Type II	通用场景
type_i	3D LCS 空间中的直接平移	强烈的全局颜色偏移
type_ii	通过双锥几何进行逐 patch 的 HSL 插值	精确的局部颜色控制

关键参数

strength（0.0–2.0）：干预强度。1.0 = 完整干预，0.0 = 无干预。
start_step / end_step：干预步骤范围。论文最优：50 步中的第 8–10 步。
mask：可选。双线性下采样至 patch 网格分辨率，用于局部控制。

工作原理

投影：将去噪预测转换到 64D patch 空间，投影到 3D LCS 基底
分解：将 3D 颜色坐标与 61D 结构残差分离
归一化：使用学习的 alpha/beta 统计量变换至参考时间步（t=50）
操作：在 3D LCS 中偏移颜色、调整色调或进行其他变换
重建：反归一化，加回保留的 61D 残差，转换回潜在空间

61D 残差（结构、纹理、细节）始终不被修改 —— 只有 3D 颜色子空间会被改变。

文件结构

ComfyUI-LCS/
├── __init__.py           # 入口（V3 + V2 兼容）
├── requirements.txt
├── core/
│   ├── calibration.py    # PCA 校准流程
│   ├── color_space.py    # 双锥 LCS ↔ HSL 映射
│   ├── defaults.py       # 论文中的 Alpha/beta 表
│   ├── lcs_data.py       # LCSData 数据类
│   ├── patchify.py       # Patch ↔ 潜在空间转换
│   └── timestep.py       # Sigma/时间步工具
├── nodes/
│   ├── calibrate.py      # LCSCalibrate, LCSLoadData
│   ├── intervene.py      # LCSColorIntervene, LCSColorBatch, LCSToneAdjust
│   └── observe.py        # LCSPreviewColors, LCSStepObserver
├── data/                 # 缓存的校准文件
└── web/js/
    └── tone_preset.js    # 前端预设同步

引用

@inproceedings{lcs2026,
  title={The Latent Color Subspace},
  author={...},
  booktitle={ICML},
  year={2026},
  note={arXiv:2603.12261v1}
}

许可证

MIT

6.0 KiB Raw Blame History Unescape Escape