ComfyUI-EasyColorCorrector 🎨
Professional-grade AI-powered color correction and enhancement for ComfyUI
[](https://opensource.org/licenses/MIT)
[](https://www.python.org/downloads/)
[](https://github.com/comfyanonymous/ComfyUI)
[!WARNING]
Release v1.2.1 introduces major refactoring and enhancements and a need name change to ComfyUI-EasyColorCorrector. This is a breaking change and you will may need to uninstall the old node before installing the new one.
[!NOTE]
Disclaimer: I’m obviously joking about the “$1000+ plugin” claims and competing with Photoshop – this is a ComfyUI custom node, not industry software! 😄 BUT it does have legit AI-powered features, face detection, and professional color science. And hey, it beats paying Adobe! 💰
🆕 Latest Updates (Major Update)
✨ Major Professional Improvements & Fixes
Manual Mode is now truly professional: Real temperature + tint separation (LAB color space), not just hue sliders
Preset Mode is transparent: Sliders update live to show exactly what each preset does
Slider ranges are realistic: No more wild values—contrast/brightness are now in sensible ranges
White Balance is professional: -1.0 (cool/blue) to +1.0 (warm/orange), like real color grading software
Skin tone processing fixed: No more green tint on light skin—portraits look natural
Smart UI: AI analysis auto-disables in Manual mode for pure manual control
Live Preset Updates: See exactly what each preset does to your sliders
2-Axis Color Control: Separate temperature and tint like the pros use
Working 3-Way Correction: Lift/Gamma/Gain actually works now
🆕 New Nodes
VAE Color Corrector: Fix VAE-induced color shifts in inpainting/img2img workflows with advanced bias analysis
Batch Color Corrector: Process entire video sequences with the same AI-powered corrections
RAW Image Processor: Direct RAW file processing with professional color science
Color Corrector Viewer: Real-time preview and analysis for video workflows
Color Palette Extractor: Standalone palette extraction for reference workflows
Film Emulation: Professional film stock emulation with highlight rolloff
The main Color Corrector is professional-grade and rock solid. The specialized nodes target specific workflows and advanced use cases.
🚀 What is this?
ComfyUI-EasyColorCorrector *is a professional-grade, AI-powered color correction suite that democratizes advanced color grading for the ComfyUI ecosystem.*
That’s the fancy version.
Real talk: I’m not trying to take over the color correction world. It’s just a node.
I built it to bridge the gap between “AI, *do the thing*” and “let me tweak this like I’m grading a Netflix series.”
If it helps artists, creators, and chaos-powered ComfyUI users get better color without paying Adobe or wiring 12 nodes together? Mission accomplished.
If not… hey, at least now you have histograms.
This node is *opinionated* about color science – it leverages computer vision, perceptual color spaces, and advanced algorithms to deliver results that rival industry-standard color grading tools.
🤯 The TL;DR
🤖 AI-Powered: Face detection, scene analysis, content-aware enhancement
🎨 Professional: LAB color space, 3-way color correction, perceptual curves
⚡ Real-time: Selective node execution for instant feedback
🎛️ Three Modes: Auto (AI magic), Preset (curated looks), Manual (full control)
✨ Key Features
🤖 AI-Powered Intelligence
Face Detection: Automatic skin tone preservation using OpenCV with advanced segmentation
Advanced Scene Analysis: Detects anime, concept art, stylized art, detailed illustrations, portraits, and realistic photos
Lighting Detection: Low-light, bright, flat, optimal lighting analysis
Content-Aware Enhancement: Specialized profiles for artistic content vs. photographic content
Edge-Aware Processing: Multi-scale edge detection with detail preservation
Advanced Segmentation: SLIC-based skin region detection for precise face enhancement
🎨 Professional Color Science
LAB Color Space: Perceptual white balance and color corrections with Bradford chromatic adaptation
K-means Analysis: Intelligent dominant color extraction in both RGB and LAB spaces
3-Way Color Corrector: Professional lift/gamma/gain controls
Perceptual Curves: Gamma-aware brightness and contrast
Advanced Color Matching: Robust reference image color matching with histogram and LAB methods
Professional Color Extraction: Brightness-sorted palette generation with perceptual spacing
⚡ Performance & Usability
Real-time Preview: Selective node execution (no full workflow runs)
Graceful Fallbacks: Works without advanced libraries installed
Three Distinct Modes: Each optimized for different use cases
Smart UI: Mode-specific controls with intelligent tooltips
Visual Outputs: Built-in histogram and color palette image generation
Multiple Output Types: Processed image, palette data, histogram image, and palette image
🎛️ The Three Modes
🤖 Auto Mode: AI Magic
Just make it look amazing
The AI analyzes your image and applies:
Scene-specific enhancement profiles for 6 content types (anime, concept art, stylized art, detailed illustrations, portraits, realistic photos)
Intelligent white balance in LAB color space
Face-aware skin tone preservation using OpenCV detection
Adaptive contrast and saturation with up to 50% boost for concept art
Lighting-aware brightness adjustments for low-light, bright, flat, and optimal conditions
Enhanced glow effects for artistic content with stronger highlight enhancement
Perfect for: Quick enhancement, artistic content, concept art, anime, batch processing
🎨 Preset Mode: Curated Looks
Give me that specific vibe
30 professional presets with intelligent slider updates:
Portrait Presets (5): Natural, Warm, Cool, High Key, Dramatic
Concept Art & Illustration (6): Epic Fantasy, Sci-Fi Chrome, Dark Fantasy, Vibrant Concept, Matte Painting, Digital Art
Artistic & Stylized (6): Anime Bright, Anime Moody, Cyberpunk, Pastel Dreams, Neon Nights, Comic Book
Cinematic (5): Cinematic, Teal & Orange, Film Noir, Vintage Film, Bleach Bypass
Natural (4): Golden Hour, Blue Hour, Sunny Day, Overcast
Classic (4): Sepia, Black & White, Faded, Moody
Enhanced Features:
Live slider updates: Presets automatically update parameter sliders for full transparency
Content-aware adaptation: Presets automatically adjust based on detected scene type
Artistic content boost: Up to 40% enhancement for concept art and illustrations
Smart variation: Intelligent randomization that respects the original artistic intent
Perfect for: Consistent styling, concept art workflows, artistic looks, client work
🎛️ Manual Mode: Professional Control
*”I know exactly what I want”*
Professional-grade tools with pure manual control:
Professional Temperature & Tint: Separate LAB color space controls (blue↔orange, green↔magenta)
3-way color corrector: Lift/Gamma/Gain with balanced ranges for precise control
Intelligent UI: AI analysis auto-disabled for pure manual workflow
Balanced parameter ranges: Realistic slider ranges that don’t break at low values
Professional color grading: Works like Lightroom/DaVinci Resolve
Film-like grain with luminance preservation
Enhanced sliders: Fine-tuned steps for precise adjustments
Perfect for: Precision work, professional color grading, learning, complex corrections
🔧 VAE Color Corrector
Fix VAE-Induced Color Shifts in Inpainting Workflows
A specialized node designed to solve a common ComfyUI problem: when images go through VAE encoding/decoding (especially in inpainting workflows), they often develop noticeable color shifts compared to the original input.
✨ Key Features
Four Correction Methods:
Luminance Zones: Professional shadows/midtones/highlights correction (recommended)
Histogram Matching: Match color distributions between original and processed
Statistical Matching: Align color statistics (mean/std) across channels
Advanced 3D LUT: Precise 3D color mapping using k-means clustering
VAE-Aware Analysis:
Analyzes color bias patterns across luminance zones
Detects high/medium/low VAE color bias automatically
Applies intelligent strength adjustments based on bias severity
Smart Preservation:
Auto-detect: Automatically identifies and preserves inpainted areas
Mask-based: Use custom masks (white=preserve, black=correct)
Edge feathering: Smooth transitions between corrected/preserved areas
Performance Features:
Lock Input: Prevents upstream reprocessing when adjusting parameters
Smart Caching: Preserves expensive VAE analysis between adjustments
Safe Color Clamping: Prevents quantization artifacts (black squares)
🎯 Perfect For
Inpainting workflows where color shifts occur after VAE processing
Img2img workflows with noticeable color deviations
Batch processing with consistent VAE color correction
Professional workflows requiring precise color matching
🚀 Usage
Connect original image (before VAE encoding)
Connect processed image (after VAE decoding)
Choose correction method (start with “luminance_zones”)
Adjust correction strength (0.8 is usually good)
Enable lock_input_image for real-time parameter tuning
Original Image → VAE Encode → [Inpainting/Processing] → VAE Decode → Processed Image
↓ ↓
└──────────────────── VAE Color Corrector ←─────────────────────────┘
↓
Color-Corrected Result
📊 Visual Outputs & Analysis
🖼️ Four Output Types
The node provides four distinct outputs for maximum workflow flexibility:
📸 Processed Image: Your enhanced/corrected image
🎨 Palette Data: Comma-separated hex color string for other nodes
📊 Histogram Image: RGB histogram visualization (512×768px)
🎨 Palette Image: Clean color swatches visualization (600×120px)
📈 Histogram Visualization
Professional RGB histogram with proper scaling and contrast
Multi-channel overlay showing red, green, and blue distributions
Dark background with subtle grid lines for better readability
Connect to Preview Image node to view during processing
🌈 Color Palette Extraction
6 dominant colors extracted using advanced K-means clustering
Perceptual LAB-space analysis for better color representation
Brightness-sorted arrangement for natural palette flow
Clean swatch design without borders or text clutter
Connect to Preview Image node for instant palette viewing
🔧 Reference Image Color Matching
Robust histogram matching prevents black spots and artifacts
Conservative LAB adjustments for natural color transitions
Multi-method approach with safe fallbacks
Strength control from 0.0 (no effect) to 1.0 (full matching)
🛠️ Installation
Quick Install (Recommended)
cd ComfyUI/custom_nodes
git clone https://github.com/regiellis/ComfyUI-EasyColorCorrector.git
cd ComfyUI-EasyColorCorrector
pip install -r requirements.txt
Manual Install
Download and extract to ComfyUI/custom_nodes/ComfyUI-EasyColorCorrector
Install dependencies:
pip install \
"torch>=1.12.0" \
"numpy>=1.21.0" \
"opencv-python>=4.8.0" \
"scikit-learn>=1.3.0" \
"scikit-image>=0.21.0" \
"colour-science>=0.4.3" \
"scipy>=1.11.0" \
"rawpy>=0.18.0" \
"imageio>=2.28.0" \
"OpenEXR>=1.3.9" \
"Pillow>=9.0.0" \
"torchvision>=0.13.0" \
"huggingface_hub>=0.15.0" \
"timm>=0.9.2"
Restart ComfyUI
Note: The node works without the advanced libraries, but you’ll miss the AI-powered features. Install them for the full experience!
🎮 Usage
Basic Workflow
Main Color Corrector
Add “Easy Color Corrector” node to your workflow
Connect an image input
Choose your mode:
Auto: Enable AI analysis and let it work its magic
Preset: Pick a style and optionally enable AI adaptation
Manual: Full control with professional tools
Connect outputs as needed:
Main image output → Save Image or further processing
Histogram output → Preview Image (to see RGB analysis)
Palette output → Preview Image (to see color swatches)
Palette data → Text nodes or other color-aware nodes
VAE Color Corrector
Add “VAE Color Corrector” node to your workflow
Connect original image (before VAE) and processed image (after VAE)
Choose correction method (start with “luminance_zones”)
Adjust correction strength (0.8 recommended)
Optionally provide a mask or enable auto-preserve for inpainted areas
📊 Enabling Visual Analysis
Enable “Extract Palette” to generate:
✅ RGB histogram visualization
✅ Color palette extraction and visualization
✅ Dominant color analysis in the console
✅ Advanced K-means clustering
When disabled, histogram and palette outputs will be black images.
🔥 Pro Tips
Enable Real-time Preview for instant feedback (uses selective execution)
Manual Mode auto-disables AI for pure manual control (can re-enable if needed)
Preset sliders update automatically showing exactly what the preset does
White Balance: -1.0 = cooler/blue, +1.0 = warmer/orange (professional range)
Temperature & Tint: Separate controls in Manual mode for professional color grading
Face detection works best with clear, well-lit portraits
Try presets first, then fine-tune with manual controls
Enable Extract Palette to see histogram and color analysis
Use Reference Image for mood board color matching
Connect histogram/palette outputs to Preview Image nodes for visual feedback
Example Console Output
🤖 AI Analysis: concept_art scene, good lighting, 0 faces detected
🎨 AI-Enhanced Preset: Epic Fantasy adapted for concept_art/good
🎛️ Manual Mode: Auto-disabled AI analysis for pure manual control
🌡️ Applied professional color adjustments: temperature: 0.3 (warmer), tint: -0.1 (green)
🎨 Extracted color palette: #2A1810,#8B4513,#D2691E,#F4A460,#DEB887,#F5DEB3
🖼️ Histogram and palette images available as separate node outputs
⚙️ Advanced Configuration
AI Analysis Control
ai_analysis: bool = True # Enable AI-powered features
adjust_for_skin_tone: bool = True # Face detection + skin preservation
white_balance_strength: 0.0 # Temperature: -1.0 (cool) to +1.0 (warm)
enhancement_strength: 0.2 # Overall AI enhancement power (balanced default)
extract_palette: bool = False # Generate histogram and palette images
Lock Input Image
lock_input_image: bool = False # Lock input to prevent upstream reprocessing
# When enabled: Caches input image so upstream nodes don't reprocess
# Result: Faster parameter adjustments without full workflow regeneration
Performance Modes
Full AI: All features enabled (default)
Basic Mode: AI analysis disabled for maximum speed
Real-time: Selective execution for instant feedback
🧠 The Science Behind It
Computer Vision Pipeline
OpenCV Face Detection: Haar cascade classifiers with SLIC segmentation
K-means Color Analysis: 5-cluster dominant color extraction in RGB and LAB spaces
Edge Density Analysis: Multi-scale edge detection for anime vs. photo classification
Histogram Analysis: Lighting condition detection and RGB distribution analysis
Advanced Segmentation: SLIC superpixel segmentation for precise skin tone detection
Color Science
LAB Color Space: Perceptual uniformity for white balance with Bradford chromatic adaptation
Perceptual Curves: Gamma-aware brightness/contrast with edge preservation
Selective Saturation: Vibrancy that preserves skin tones using advanced masking
3-Way Correction: Professional lift/gamma/gain masking with artistic content awareness
Advanced Color Matching: Histogram-based and LAB color space reference matching
Professional Palette Extraction: Brightness-sorted dominant colors with perceptual spacing
AI Enhancement Profiles
if scene_type == "concept_art":
contrast_boost = 0.35 * enhancement_strength
saturation_boost = 0.5 * enhancement_strength
elif scene_type == "anime":
contrast_boost = 0.25 * enhancement_strength
saturation_boost = 0.45 * enhancement_strength
elif scene_type == "portrait":
# More conservative for faces
contrast_boost = 0.15 * enhancement_strength
# + automatic skin warmth boost
🤝 Contributing
Found a bug? Have an idea? Want to add more presets?
Fork the repository
Create a feature branch (git checkout -b amazing-feature)
Commit your changes (git commit -m 'Add amazing feature')
Push to the branch (git push origin amazing-feature)
Open a Pull Request
🎨 Adding New Presets
"Your Preset Name": {
"warmth": 0.1, # -1.0 to 1.0
"vibrancy": 0.3, # -1.0 to 1.0
"contrast": 0.2, # -1.0 to 1.0
"brightness": 0.05 # -1.0 to 1.0
}
📝 License
This project is licensed under the MIT License – see the LICENSE file for details.
🙏 Acknowledgments
ComfyUI Community for the amazing framework
OpenCV for computer vision capabilities
Colour Science for professional color space operations
scikit-learn for machine learning algorithms
💡 FAQ
Q: Do I need all the dependencies?
A: The node works without them, but you’ll miss the AI features. Install them for the full experience!
Q: Is this better than Photoshop?
A: For color correction? It’s competitive. For everything else? No. 😄
Q: Can I use this commercially?
A: Yes! MIT license means you can use it however you want.
Q: Why is it called “Easy” when it’s so advanced?
A: Because the AI makes professional color grading *easy*. The complexity is hidden behind intelligent automation.
Q: What’s the difference between the main node and VAE Color Corrector?
A: The main node is for general color correction and enhancement. VAE Color Corrector specifically fixes color shifts that occur when images go through VAE encoding/decoding in ComfyUI workflows.
Q: When should I use the VAE Color Corrector?
A: Use it when you notice color shifts after inpainting, img2img, or any workflow where images pass through a VAE. Connect the original (pre-VAE) and processed (post-VAE) images.
Q: Why am I getting black squares in my images?
A: This was a quantization artifact issue that’s now fixed with safe color clamping. Update to the latest version for the fix.
Made with ❤️ and way too much caffeine
*Author: Regi E.*
*”Making AI-powered color grading accessible to everyone”*
