"src/sdk/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "7108466c214186769ab16da80636ffc202535284"
Commit c16f5744 authored by comfyanonymous's avatar comfyanonymous
Browse files

Fix SplitImageWithAlpha and JoinImageWithAlpha.

parent 1f2f4eaa
...@@ -158,7 +158,7 @@ class SplitImageWithAlpha: ...@@ -158,7 +158,7 @@ class SplitImageWithAlpha:
def split_image_with_alpha(self, image: torch.Tensor): def split_image_with_alpha(self, image: torch.Tensor):
out_images = [i[:,:,:3] for i in image] out_images = [i[:,:,:3] for i in image]
out_alphas = [i[:,:,3] if i.shape[2] > 3 else torch.ones_like(i[:,:,0]) for i in image] out_alphas = [i[:,:,3] if i.shape[2] > 3 else torch.ones_like(i[:,:,0]) for i in image]
result = (torch.stack(out_images), torch.stack(out_alphas)) result = (torch.stack(out_images), 1.0 - torch.stack(out_alphas))
return result return result
...@@ -180,7 +180,7 @@ class JoinImageWithAlpha: ...@@ -180,7 +180,7 @@ class JoinImageWithAlpha:
batch_size = min(len(image), len(alpha)) batch_size = min(len(image), len(alpha))
out_images = [] out_images = []
alpha = resize_mask(alpha, image.shape[1:]) alpha = 1.0 - resize_mask(alpha, image.shape[1:])
for i in range(batch_size): for i in range(batch_size):
out_images.append(torch.cat((image[i][:,:,:3], alpha[i].unsqueeze(2)), dim=2)) out_images.append(torch.cat((image[i][:,:,:3], alpha[i].unsqueeze(2)), dim=2))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment