fix: output bbox-cropped image instead of mask-applied image
This commit is contained in:
@@ -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 {
|
||||
|
||||
Reference in New Issue
Block a user