This commit is contained in:
Patrick Morgan
2024-12-15 21:11:59 -08:00
parent 3198afc524
commit 93d4022b09
+18 -1
View File
@@ -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