diff --git a/py/nodes/mask_canvas_editor.py b/py/nodes/mask_canvas_editor.py index 4f0cb5f..c08b257 100644 --- a/py/nodes/mask_canvas_editor.py +++ b/py/nodes/mask_canvas_editor.py @@ -248,10 +248,7 @@ class MaskCanvasEditor: sampled = sampled.squeeze(0).permute(1, 2, 0) - mask_expanded = m_cropped.unsqueeze(-1) - sampled_masked = sampled * mask_expanded - - cropped_images.append(sampled_masked) + cropped_images.append(sampled) cropped_masks.append(m_cropped) return {