Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

(minor) Tweak order of mask operations.

Browse Source
This commit is contained in:
Ryan Dick 2024-08-01 10:12:24 -04:00
parent c3a6a6fb22
commit e6a512aa86
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
invokeai/app/invocations/mask.py
View File

@ -138,8 +138,7 @@ class MaskTensorToImageInvocation(BaseInvocation, WithMetadata, WithBoard):
# Ensure that the mask is binary. # Ensure that the mask is binary.
if mask.dtype != torch.bool: if mask.dtype != torch.bool:
mask = mask > 0.5 mask = mask > 0.5
mask_np = mask.float().cpu().detach().numpy() * 255 mask_np = (mask.float() * 255).byte().cpu().numpy()
mask_np = mask_np.astype(np.uint8)
mask_pil = Image.fromarray(mask_np, mode="L") mask_pil = Image.fromarray(mask_np, mode="L")
image_dto = context.images.save(image=mask_pil) image_dto = context.images.save(image=mask_pil)