Add ComfyUI support
my bad y'all
This commit is contained in:
parent
b9e4a2cfa1
commit
78a01e432e
5 changed files with 59 additions and 16 deletions
14
mmaker_color_enhance_core.py
Normal file
14
mmaker_color_enhance_core.py
Normal file
|
|
@ -0,0 +1,14 @@
|
|||
import numpy as np
|
||||
import skimage.color
|
||||
from PIL import Image
|
||||
import imageio.core.util
|
||||
|
||||
imageio.core.util._precision_warn = lambda *args, **kwargs: None
|
||||
|
||||
def color_enhance(arr, strength: float = 1) -> Image.Image:
|
||||
lch = skimage.color.lab2lch(lab=skimage.color.rgb2lab(rgb=np.array(arr, dtype=np.uint8)))
|
||||
lch[:, :, 1] *= 100/(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))
|
||||
|
||||
def lerp(a: float, b: float, t: float) -> float:
|
||||
return (1 - t) * a + t * b
|
||||
Loading…
Add table
Add a link
Reference in a new issue