From 05b4c3c4b76ef4375d8ec7eec23208d287cd4279 Mon Sep 17 00:00:00 2001 From: Kevin Ramirez Date: Fri, 21 Jun 2024 06:01:13 +0000 Subject: [PATCH] Add support for hex format and clip rgb format --- nodes/image_nodes.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/nodes/image_nodes.py b/nodes/image_nodes.py index 8f4732c..4d2a320 100644 --- a/nodes/image_nodes.py +++ b/nodes/image_nodes.py @@ -872,7 +872,11 @@ nodes for example. mask_adjusted = mask * mask_opacity 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(', '))) + if ',' in mask_color: + color_list = np.clip([int(channel) for channel in mask_color.split(',')], 0, 255) # RGB format + else: + mask_color = mask_color.lstrip('#') + color_list = [int(mask_color[i:i+2], 16) for i in (0, 2, 4)] # Hex format mask_image[:, :, :, 0] = color_list[0] / 255 # Red channel mask_image[:, :, :, 1] = color_list[1] / 255 # Green channel mask_image[:, :, :, 2] = color_list[2] / 255 # Blue channel