stake
This commit is contained in:
+18
-1
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user