git/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2026-03-04 14:09:03 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add safety check for CDF normalization in histogram matching

Browse Source 
Co-authored-by: dunkeroni <3298737+dunkeroni@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot] 2025-10-08 17:59:59 +00:00 committed by psychedelicious
parent b1f7e2dfdc
commit 07a2369105
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/invocations/image.py
View File

@ -674,9 +674,11 @@ class ColorCorrectInvocation(BaseInvocation, WithMetadata, WithBoard):
source_cdf = source_hist.cumsum()
reference_cdf = reference_hist.cumsum()
# Normalize CDFs
source_cdf = source_cdf / source_cdf[-1]
reference_cdf = reference_cdf / reference_cdf[-1]
# Normalize CDFs (avoid division by zero)
if source_cdf[-1] > 0:
source_cdf = source_cdf / source_cdf[-1]
if reference_cdf[-1] > 0:
reference_cdf = reference_cdf / reference_cdf[-1]
# Create lookup table using linear interpolation
lookup_table = numpy.interp(source_cdf, reference_cdf, numpy.arange(256))