From c861cd6baf531e2e52ae2f8562731dd5f113c7bc Mon Sep 17 00:00:00 2001 From: cubiq Date: Fri, 30 Aug 2024 14:34:46 +0200 Subject: [PATCH] lease alpha to imageremovebg --- image.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/image.py b/image.py index f5cea3c..d2ee3e3 100644 --- a/image.py +++ b/image.py @@ -842,8 +842,9 @@ class ImageRemoveBackground: output = torch.stack(output, dim=0) output = output.permute([0, 2, 3, 1]) mask = output[:, :, :, 3] if output.shape[3] == 4 else torch.ones_like(output[:, :, :, 0]) + # output = output[:, :, :, :3] - return(output[:, :, :, :3], mask,) + return(output, mask,) """ ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~