xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-09-10 12:56:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix mask_color widget in ImageAndMaskPreview node

Browse Source
This commit is contained in:
Kevin Ramirez 2024-06-21 03:34:49 +00:00
parent 2fb0ee4934
commit 1f03079338
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes/image_nodes.py
View File

@ -873,9 +873,9 @@ nodes for example.
mask_image = mask.reshape((-1, 1, mask.shape[-2], mask.shape[-1])).movedim(1, -1).expand(-1, -1, -1, 3).clone() mask_image = mask.reshape((-1, 1, mask.shape[-2], mask.shape[-1])).movedim(1, -1).expand(-1, -1, -1, 3).clone()
color_list = list(map(int, mask_color.split(', '))) color_list = list(map(int, mask_color.split(', ')))
mask_image[:, :, :, 0] = color_list[0] // 255 # Red channel mask_image[:, :, :, 0] = color_list[0] / 255 # Red channel
mask_image[:, :, :, 1] = color_list[1] // 255 # Green channel mask_image[:, :, :, 1] = color_list[1] / 255 # Green channel
mask_image[:, :, :, 2] = color_list[2] // 255 # Blue channel mask_image[:, :, :, 2] = color_list[2] / 255 # Blue channel
preview, = ImageCompositeMasked.composite(self, image, mask_image, 0, 0, True, mask_adjusted) preview, = ImageCompositeMasked.composite(self, image, mask_image, 0, 0, True, mask_adjusted)
if pass_through: if pass_through: