ComfyUI_Gear:面向 VFX 的专业级 HDR 调色与 EXR 工作流节点

插件6个月前发布 小马良
209 0

ComfyUI_Gear 是一套专为视觉特效(VFX)和高动态范围(HDR)工作流设计的 ComfyUI 自定义节点包。它填补了 ComfyUI 在专业色彩管理领域的空白,提供了从 LogC3 解码到 ACEScct 交互式调色的完整解决方案,并支持将结果保存为工业标准的 OpenEXR 格式。

  • GitHub:https://github.com/oumad/ComfyUI_Gear

该工具包由 exr-viewer 驱动,允许用户在浏览器中通过 WebGL 进行实时、无损的 HDR 调色预览,无需反复运行整个工作流。

ComfyUI_Gear:面向 VFX 的专业级 HDR 调色与 EXR 工作流节点

核心功能

1. 交互式 ACEScct 调色面板

  • 实时预览:基于 WebGL2 的模态弹窗界面,提供色轮、Lift/Gamma/Gain/Offset 控制、示波器等专业工具。调整滑块时,画面即时更新,无需重新生成图像。
  • A|B 对比:支持连接参考图像(SDR),通过拖动擦拭线(Wipe)进行前后对比,特别适合批处理帧的同步检查。
  • LUT 支持:支持加载任意 .cube LUT 文件,内置多款电影级预设 LUT。
  • 色调映射器:内置 AgX, ACES Fitted, Hable, Reinhard 等多种色调映射算法,确保 HDR 到 SDR 转换的色彩准确性。
  • 状态持久化:调色参数会自动保存,即使关闭弹窗或重启工作流,修改后的状态依然保留。

2. LogC3 解码与 EXR 输出

  • HDR 还原:将 LoRA 或其他模型输出的 LogC3 压缩图像(范围 [0,1])逆变换为场景线性 HDR 数据(值可超过 1.0)。
  • Float16 EXR 保存:直接输出高质量的 float16 OpenEXR 文件,保留完整动态范围,方便后续在 Nuke、DaVinci Resolve 等专业软件中进行合成与调色。
  • 智能文件名:支持 ComfyUI Token(如 %year%, %batch_num%),自动添加计数后缀,防止文件覆盖。

3. 批处理与 Scrubber

  • 帧序列支持:支持输入 [N, H, W, C] 格式的批量图像张量,面板底部提供帧 Scrubber,可逐帧检查调色效果。
  • 同步切换:在批处理模式下,EXR 和 SDR 参考图会随帧滚动同步切换。
ComfyUI_Gear:面向 VFX 的专业级 HDR 调色与 EXR 工作流节点

快速开始

安装

方式一:ComfyUI Manager(推荐)

  1. 打开 ComfyUI Manager。
  2. 搜索 ComfyUI_Gear。
  3. 点击 Install 并重启 ComfyUI。

方式二:手动安装

cd ComfyUI/custom_nodes
git clone https://github.com/oumad/ComfyUI_Gear
cd ComfyUI_Gear
pip install -r requirements.txt

重启 ComfyUI。无需额外配置,exr-viewer 的前端资源已内置。

使用指南

1. LogC3 Decode + Save EXR

  • 输入:连接 HDR LoRA 或模型输出的 IMAGE(LogC3 编码,范围 0-1)。
  • 输出:
    • hdr_linear:场景线性 HDR 张量,可连接至调色节点。
    • tonemapped_preview:用于 UI 显示的快速 Reinhard 色调映射预览。
    • exr_paths:保存的 EXR 文件路径列表。

2. Color Grade (exr-viewer)

  • 连接:将 hdr_linear 连接至节点的 image 输入。可选连接 sdr_reference 用于 A|B 对比。
  • 运行:执行一次工作流以生成初始图像。
  • 调色:点击节点上的 Open grade panel 按钮。
    • 在弹出的窗口中调整色轮、曲线或加载 LUT。
    • 点击 Save (✓) 保存调色结果;点击 Cancel (×) 或按 Esc 丢弃更改。
  • 输出:
    • graded_display:经过色调映射和 OETF 转换后的 sRGB 图像(0-1),用于预览。
    • graded_linear:调色后的场景线性 HDR 数据,可连接回 EXR 保存节点以输出最终成果。

技术亮点

  • 双引擎同步:调色计算同时在前端(GLSL 片段着色器,用于实时预览)和后端(PyTorch,用于最终输出)运行,确保预览与最终结果完全一致。
  • 模块化架构:添加新节点只需编辑单个文件,便于开发者扩展功能。
  • 跨平台兼容:基于纯浏览器代码的 exr-viewer,在 Windows、Linux、macOS 上均能完美运行,只要浏览器支持 WebGL2。

系统要求

  • ComfyUI: 2024+ 近期版本
  • Python: 3.10+
  • GPU: 支持 ComfyUI 运行的任何 GPU;调色面板依赖浏览器的 WebGL2 支持。
  • 操作系统: Windows, Linux, macOS
© 版权声明

相关文章