wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-26 22:30:19 +08:00
Code Issues Actions 12 Packages Projects Releases Wiki Activity

Add support for color names

Browse Source 
see https://pillow.readthedocs.io/en/stable/_modules/PIL/ImageColor.html for supported colors
This commit is contained in:
missionfloyd 2023-04-08 21:01:44 -06:00 committed by GitHub
parent b6d1bfdc8b
commit 23a93b40f3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_post_processing.py
View File

@ -186,7 +186,7 @@ class QuantizePalette:
raise ValueError("Palette is empty")
def parse_palette(color_str):
if re.match(r'#[a-fA-F0-9]{6}', color_str):
if re.match(r'#[a-fA-F0-9]{6}', color_str) or color_str.lower() in ImageColor.colormap:
return ImageColor.getrgb(color_str)
color_rgb = re.match(r'\(?(\d{1,3}),(\d{1,3}),(\d{1,3})\)?', color_str)
@ -197,7 +197,7 @@ class QuantizePalette:
pal_img = Image.new('P', (1, 1))
pal_colors = palette.replace(" ", "")
pal_colors = re.findall(r'#[a-fA-F0-9]{6}|\(?\d{1,3},\d{1,3},\d{1,3}\)?', pal_colors)
pal_colors = re.findall(fr'#[a-fA-F0-9]{{6}}|\(?\d{{1,3}},\d{{1,3}},\d{{1,3}}\)?|{"|".join(ImageColor.colormap.keys())}', pal_colors)
pal_colors = list(itertools.chain.from_iterable(map(parse_palette, pal_colors)))
pal_img.putpalette(pal_colors)