wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-16 08:22:36 +08:00
Code Issues Actions 15 Packages Projects Releases Wiki Activity

Case-insensitive color names

Browse Source
This commit is contained in:
missionfloyd 2023-04-09 13:54:13 -06:00 committed by GitHub
parent 23a93b40f3
commit 29bbe8e2e4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_post_processing.py
View File

@ -163,7 +163,7 @@ class QuantizePalette:
return { return {
"required": { "required": {
"image": ("IMAGE",), "image": ("IMAGE",),
"palette": ("STRING", {"default": "#000000, #ffffff", "multiline": True}), "palette": ("STRING", {"default": "black, white", "multiline": True}),
"dither": (["none", "floyd-steinberg"],), "dither": (["none", "floyd-steinberg"],),
}, },
} }
@ -197,7 +197,7 @@ class QuantizePalette:
pal_img = Image.new('P', (1, 1)) pal_img = Image.new('P', (1, 1))
pal_colors = palette.replace(" ", "") pal_colors = palette.replace(" ", "")
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 = re.findall(fr'#[a-fA-F0-9]{{6}}|\(?\d{{1,3}},\d{{1,3}},\d{{1,3}}\)?|{"|".join(ImageColor.colormap.keys())}', pal_colors, re.IGNORECASE)
pal_colors = list(itertools.chain.from_iterable(map(parse_palette, pal_colors))) pal_colors = list(itertools.chain.from_iterable(map(parse_palette, pal_colors)))
pal_img.putpalette(pal_colors) pal_img.putpalette(pal_colors)