- 新增Color Match (UTK)节点:支持6种颜色匹配算法,用于图像间色彩转移 - 新增Color To Mask (UTK)节点:根据RGB颜色值创建掩码,支持阈值调节 - 新增Separate Masks (UTK)节点:分离连通组件为独立掩码,支持3种输出模式 - 新增Bbox Visualize (UTK)节点:在图像上绘制边界框,支持xywh和xyxy格式 - 重构mask分类架构:创建独立py文件封装,与image分类保持一致 - 优化Color Match节点输入顺序:image_target在前,避免bypass节点传递错误 - 完善节点分类和导航:所有新节点正确显示在右侧导航面板 - 增强依赖管理:添加color-matcher、scipy等必要依赖
3.3 KiB
3.3 KiB
Color Match (UTK) 节点使用说明
概述
Color Match (UTK) 节点是从 kjnodes 项目移植过来的颜色匹配功能,现在已成功集成到 ComfyUI Universal Toolkit 的 image 分类中。该节点能够将参考图像的颜色特征转移到目标图像上,非常适合用于自动色彩分级、照片后期处理和影片色彩统一等场景。
功能特点
- 多种颜色匹配算法:支持 6 种不同的颜色匹配方法
- 强度控制:可调节颜色匹配的强度(0.0-10.0)
- 批处理支持:支持批量图像处理
- 多线程优化:可选择开启多线程以提升处理速度
- 高质量结果:基于学术研究的先进算法
安装要求
在使用此节点之前,需要安装 color-matcher 依赖库:
pip install color-matcher
节点参数
必需参数
- image_ref:参考图像,作为颜色匹配的目标样式
- image_target:目标图像,需要被调整颜色的图像
- method:颜色匹配方法,可选择:
mkl:Monge-Kantorovich Linearization(默认)hm:Histogram Matchingreinhard:Reinhard et al. methodmvgd:Multi-Variate Gaussian Distributionhm-mvgd-hm:Histogram Matching + MVGD + Histogram Matchinghm-mkl-hm:Histogram Matching + MKL + Histogram Matching
可选参数
- strength:颜色匹配强度(默认:1.0,范围:0.0-10.0)
- multithread:是否启用多线程处理(默认:True)
使用方法
- 在 ComfyUI 中,在节点菜单中找到
UTK/image分类 - 添加
Color Match (UTK)节点 - 连接参考图像到
image_ref输入 - 连接目标图像到
image_target输入 - 选择合适的颜色匹配方法
- 调整强度参数(可选)
- 运行工作流
算法说明
推荐的方法选择
- mkl:适合大多数场景的通用方法,效果平衡
- hm-mvgd-hm:复合方法,通常能获得最佳效果
- reinhard:经典方法,适合艺术风格转换
- hm:简单快速,适合基本的色彩调整
性能优化
- 对于单张图像,建议关闭多线程
- 对于批量处理,建议开启多线程以提升速度
- 较低的强度值(0.3-0.7)通常能产生更自然的效果
技术参考
该节点基于以下学术研究:
- Reinhard et al. 的颜色转移方法
- Pitie et al. 提出的 Monge-Kantorovich Linearization
- Multi-Variate Gaussian Distribution 转移的解析解
项目参考:https://github.com/hahnec/color-matcher/
故障排除
常见问题
- 导入错误:确保已安装
color-matcher库 - 处理失败:检查输入图像是否为有效的张量格式
- 内存不足:对于大图像,可以尝试关闭多线程或降低批处理大小
错误处理
节点包含完善的错误处理机制:
- 如果颜色匹配失败,会自动返回原始图像
- 所有错误信息会在控制台中显示
- 支持优雅降级,确保工作流不会中断
更新日志
- v1.0:初始版本,从 kjnodes 移植并集成到 UTK
- 支持所有原始功能和参数
- 添加了详细的文档和错误处理
- 优化了代码结构和性能
此节点是 ComfyUI Universal Toolkit 的一部分,致力于为 ComfyUI 用户提供更丰富的图像处理功能。