mmaker/sd-webui-color-enhance
1
Fork 0
Code Issues Pull requests Activity

Add ComfyUI support

Browse source 
my bad y'all
This commit is contained in:
MMaker 2023-11-15 19:58:17 -05:00
commit 78a01e432e
Signed by: mmaker
GPG key ID: CCE79B8FEDA40FB2
5 changed files with 59 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

18
scripts/color_enhance.py
View file

@ -1,14 +1,8 @@
import gradio as gr
import imageio.core.util
import numpy as np
import skimage.color
from PIL import Image
from modules import scripts_postprocessing
from modules.ui_components import FormRow
imageio.core.util._precision_warn = lambda *args, **kwargs: None
import mmaker_color_enhance_core as lib
class ScriptPostprocessingColorEnhance(scripts_postprocessing.ScriptPostprocessing):
@ -25,14 +19,8 @@ class ScriptPostprocessingColorEnhance(scripts_postprocessing.ScriptPostprocessi
return
info_bak = {} if not hasattr(pp.image, "info") else pp.image.info
pp.image = self._color_enhance(pp.image, strength)
pp.image = lib.color_enhance(pp.image, strength)
pp.image.info = info_bak
pp.info["Color Enhance"] = strength
def _lerp(self, a: float, b: float, t: float) -> float:
return (1 - t) * a + t * b
def _color_enhance(self, arr, strength: float = 1) -> Image.Image:
lch = skimage.color.lab2lch(lab=skimage.color.rgb2lab(rgb=np.array(arr, dtype=np.uint8)))
lch[:, :, 1] *= 100/(self._lerp(100, lch[:, :, 1].max(), strength)) # Normalize chroma component
return Image.fromarray(np.array(skimage.color.lab2rgb(lab=skimage.color.lch2lab(lch=lch)) * 255, dtype=np.uint8))