From 93d4022b097f847ff1424a88b50b46bedca6a0dc Mon Sep 17 00:00:00 2001 From: Patrick Morgan Date: Sun, 15 Dec 2024 21:11:59 -0800 Subject: [PATCH] stake --- nodes/nodes.py | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/nodes/nodes.py b/nodes/nodes.py index ac81180..990afbd 100644 --- a/nodes/nodes.py +++ b/nodes/nodes.py @@ -333,7 +333,6 @@ class MaskContourProcessor: Returns: ndarray: Updated mask with rendered effect """ - # Create a temporary canvas for anti-aliased drawing height, width = mask_array.shape canvas = np.zeros((height, width), dtype=np.float32) @@ -375,6 +374,12 @@ class MaskContourProcessor: canvas[rr, cc] = np.maximum(canvas[rr, cc], val) + # Debug: Check if canvas has been modified + if np.any(canvas > 0): + print("Canvas has been modified.") + else: + print("Canvas is still empty.") + # Combine with original mask return np.maximum(mask_array, canvas) @@ -436,10 +441,22 @@ class MaskContourProcessor: ) effects_data.append(effect) + # Debug: Check if effects_data is being generated + if not effects_data: + print("No effects data generated.") + else: + print(f"Generated {len(effects_data)} effects.") + # Render all effects for effect in effects_data: output_mask = self.render_effect_to_mask(output_mask, effect) + # Debug: Check if output_mask has been modified + if np.any(output_mask != mask_np): + print("Output mask has been modified.") + else: + print("Output mask is unchanged.") + return (torch.from_numpy(output_mask),) # Node class mappings