The Ultimate AI Auto Color Correction! How to Use ComfyUI-SuperBeasts

Introduction

Hello, I’m Easygoing.

In this post, we’ll explore color correction for illustrations generated by image generation AI.

sRGB and LAB Color Spaces

Before diving into color correction, let’s take a look at color spaces.

sRGB (Standard RGB) Color Space

Monitors on computers and smartphones display colors by combining three primary colors: Red, Green, and Blue.

• Red: 0–255
• Green: 0–255
• Blue: 0–255

The 0–255 range corresponds to 8 bits in binary, and the total 24-bit RGB color representation is known as standard RGB or True Color.

sRGB is designed primarily for monitor display on smartphones and computers.

LAB Color Space

In contrast, the LAB color space was defined in 1976 by the International Commission on Illumination (CIE) based on human perception.

The LAB color space consists of three components:

• L (Lightness): Brightness of the color, ranging from 0 (black) to 100 (white).
• a (Red-Green Axis): Redness (positive values) to greenness (negative values), typically -128 to +127.
• b (Yellow-Blue Axis): Yellowness (positive values) to blueness (negative values), typically -128 to +127.

Since the LAB color space is based on human perception, the same LAB values represent the same color across different devices.

PNG files handled by image generation AI use sRGB, but by converting to LAB for adjustments and then reverting to sRGB, you can achieve natural color adjustments aligned with human perception.

Comparing Three Types of Color Adjustments!

Let’s compare three types of color adjustments:

• RGB Correction
• LAB Correction
• AI Auto Correction

Original Illustration

We’ll be correcting the following illustration generated by an SDXL anime model.

This illustration has clean color usage overall but appears slightly bluish, and due to the traditional e-prediction model of SDXL, it feels low in contrast and whitish.

Let’s correct the colors of this illustration.

1. RGB Correction

First, let’s try RGB correction using the LayerColor: AutoAdjust V2 node from ComfyUI-LayerStyle.

RGB correction increases contrast, emphasizing the black in the clothing and giving the image a tighter feel. However, the skin tone balance shifts, making the face darker than the original.

2. LAB Correction

Next, let’s try color correction in LAB space using the HDR Effects node from ComfyUI-SuperBeasts.

This correction is stronger than RGB correction but maintains the balance of skin tones.

3. AI Auto Correction

Finally, let’s try AI auto correction using the Super Pop Color Adjustment node from ComfyUI-SuperBeasts.

The AI prioritizes skin tone adjustments, removing the dull bluish tone and resulting in a healthy-looking illustration.

AI Auto Correction Is Convenient!

With standard RGB correction, manual adjustments per image can yield optimal results, but incorporating it into a ComfyUI workflow for automatic adjustments can significantly alter the balance of some illustrations, limiting how strong the correction can be.

In contrast, AI auto color correction optimizes adjustments for each illustration, allowing for fully automated processing within a workflow.

Additionally, performing AI correction followed by LAB correction allows you to fine-tune the colors to your desired expression.

AI Auto Correction → LAB Correction Workflow

Here’s the workflow for AI auto correction followed by LAB correction.

flowchart LR

subgraph SDXL
A1(Original Illustration)
end

subgraph Super POP
B1(Skin Tone Adjustment)
end

subgraph HDR Effects
C1(Enhance Contrast)
end

subgraph HiDream
D1(Redraw)
end

A1-->B1
B1-->C1
C1-->D1

1. Generate the original illustration with an SDXL anime model.
2. Adjust skin tones with AI auto correction.
3. Enhance contrast with LAB correction.
4. Redraw with HiDream.

Actual Illustrations!

Let’s take a look at the actual illustrations.

SDXL (Original)

This is the original illustration generated by the SDXL anime model.

AI Auto Correction with Super POP

The original illustration had a slightly dull bluish tone, but the colors are nicely corrected, especially improving the skin’s complexion.

LAB Correction with HDR Effects

Next, we apply color correction in LAB space using the HDR Effects node.

The SDXL model struggles with high-contrast expressions, but by lowering the level of dark colors, we achieve a tighter overall expression. Additionally, reducing the whitish tone helps mitigate the noise typical of diffusion models in image generation AI.

Redrawing with HiDream Model

Finally, we redraw the illustration using the high-quality HiDream-I1-Dev model to complete it.

While the HiDream model generates high-quality illustrations, its skin tone expression is not always ideal. By using color-adjusted illustrations from SDXL, we can achieve preferred color expressions even with the HiDream model.

Other Examples!

Here are other illustrations generated using the same workflow.

Luxurious House

Stylish Bar

Nighttime Amusement Park

Workflow

Here’s the color adjustment workflow used.

1. Color Adjustment Workflow

• SuperBeastsAI_Super_POP_HDR_Effects_20250801.json

In environments with GPU support, setting the SB Load Model device to GPU enables faster processing.

2. SDXL -> Color Adjustment -> HiDream Workflow

This is the workflow used to generate the illustrations.

• SDXL_Color_adjust_HiDream_20250801.json

Models Used

• SDXL
  • anima_pencil-v5.0.0
• HiDream Models
  • HiDream
    • HiDream_I1_Dev_BF16
  • Text Encoders
    • Llama-3.1-8b-instruct-BF16
    • flan-t5-xxl_TE-only_FP32
    • CLIP-ViT-bigG-14-laion2B-39B-b160k-FP32
    • CLIP-SAE-ViT-L-14-FP32
  • VAE
    • FLUX1-schnell-AE-FP32
• AI Upscale Model
  • IllustrationJaNai_V1_DAT2_190k (Not for commercial use)

About the HiDream Model

About the AI Upscale Model

The AI upscale model used in the workflow, IllustrationJaNai_V1_DAT2_190k, is not for commercial use. Please use a different model for commercial illustrations.

Custom Nodes

Here are the custom nodes used in this workflow.

ComfyUI-LayerStyle

ComfyUI-SuperBeasts

Summary: AI Auto Color Correction Is Convenient!

• AI auto color correction is convenient.
• LAB correction provides natural results.
• HiDream responds effectively to color prompts.

Color expression is a critical element in AI illustrations.

Color adjustment involves many technical terms and can be challenging, but with the addition of the AI auto color correction node in ComfyUI-SuperBeasts, even beginners can easily adjust colors.

ComfyUI is highly extensible, with amazing custom nodes being released continuously, making it a great learning experience.

I look forward to continuing to pursue original expressions through ComfyUI.

Thank you for reading until the end!