ComfyUI Extension: eric-color-correction

Authored by EricRollei

Created

Updated

3 stars

Run ComfyUI workflows without the setup

No installs, no CUDA version roulette, no GPU sitting idle on your bill. Bring a workflow and run it in the browser.

Professional color correction and skin masking nodes for ComfyUI

README

Eric Color Correction for ComfyUI

Professional color correction, skin masking, and batch utility nodes for ComfyUI.

This custom node pack focuses on practical skin-tone workflows:

  • Analyze skin line angle and luminance against targets.
  • Remove color cast with skin-guided white balance.
  • Correct skin hue and luminance selectively.
  • Build masks from face parsing, human parsing, and MediaPipe landmarks.
  • Reuse correction values across image batches.

Highlights

  • Production-oriented tools for people-centric imagery.
  • Works with batch image pipelines.
  • Supports workflow combinations with custom masks.
  • Includes multiple segmentation/masking paths for different quality and speed tradeoffs.

Included Nodes

Analysis and Correction

  • ColorCorrectionAnalyzer
  • ColorCastRemover
  • SkinToneCorrector
  • SkinLuminanceAdjust

Batch Tools

  • MaskBatchPropagate
  • ApplySavedCorrection

Masking

  • FaceParsingSkinMask
  • HumanParsingSkinMask
  • MediaPipeSkinMask

Requirements

Core runtime requirements:

  • torch
  • numpy
  • opencv-python
  • mediapipe
  • transformers

See requirements.txt for baseline dependencies and pyproject.toml for packaging metadata.

Installation

Option 1: ComfyUI Manager

  1. Open ComfyUI Manager.
  2. Search for this node pack.
  3. Install and restart ComfyUI.

Option 2: Manual Install

  1. Clone this repository into your ComfyUI custom_nodes directory.
  2. Install dependencies:
    • pip install -r requirements.txt
  3. Restart ComfyUI.

Example custom_nodes path:

  • ComfyUI/custom_nodes/Eric_Color_Correction

Quick Workflow Pattern

  1. Generate or load an image.
  2. Build a skin mask with one of:
    • FaceParsingSkinMask
    • HumanParsingSkinMask
    • MediaPipeSkinMask
  3. Run ColorCorrectionAnalyzer to inspect skin line deviation and luminance.
  4. Apply ColorCastRemover for white balance cleanup.
  5. Apply SkinToneCorrector and/or SkinLuminanceAdjust.
  6. Re-check with ColorCorrectionAnalyzer.
  7. For batch pipelines, reuse values with ApplySavedCorrection.

Example Workflows

Example workflow templates are included in the examples folder.

Workflow Overview

Example ComfyUI graph:

Example workflow

Example Results and Tool Views

Color cast removal example:

Color cast removal example

Skin tone adjustment examples:

Skin tone adjustments

Face parsing and skin parsing comparison:

Face parsing and skin parsing

Vectorscope information example:

Vectorscope info

  • examples/workflows/portrait_face_parsing_api.json
  • examples/workflows/portrait_mediapipe_api.json
  • examples/workflows/two_image_batch_mask_propagation_api.json
  • examples/workflows/reference_transfer_saved_correction_api.json

See examples/README.md for usage notes and placeholder filenames.

Troubleshooting

Missing model or import errors

  • Confirm required dependencies are installed.
  • Restart ComfyUI after dependency changes.

Slow first run

  • Some parsing nodes load models on first execution.
  • Subsequent runs are typically faster due to model caching.

CUDA memory issues

  • Reduce batch sizes.
  • Use lower-resolution previews for correction tuning.

Versioning and Releases

  • Keep release notes in CHANGELOG.md.
  • Tag releases using semantic versioning where possible.
  • Include screenshots or example workflows in release notes for better discoverability.

Publishing Checklist (GitHub)

  • Ensure .gitignore excludes local runtime and sensitive files.
  • Confirm README, LICENSE, and dependency documentation are up to date.
  • Verify pyproject.toml metadata (name, version, contacts, URLs).
  • Run a quick import smoke test before creating a release tag.

License

This project uses a dual-license model.

  • Non-commercial use: Modified CC BY-NC 4.0 with additional attribution requirements.
  • Commercial use: Requires a separate commercial license.

See LICENSE for full terms and LICENSE-DEPENDENCIES.md for third-party license notes.

Contact

Run ComfyUI workflows without the setup

No installs, no CUDA version roulette, no GPU sitting idle on your bill. Bring a workflow and run it in the browser.

Learn more