Fix ImageTransformCropCorners with improved alpha channel handling and antialiasing

2023-11-24 14:52:37 +03:00
parent 60c1ae72a9
commit 066edad319
1 changed files with 1 additions and 1 deletions
@@ -231,7 +231,7 @@ class ImageTransformCropCorners:

        def crop_tensor(tensor):
            return torch.stack([
-                tensor[:, :, i] - mask for i in range(tensor.shape[2])
+                (tensor[:, :, i] - mask).clamp(0, 1) for i in range(tensor.shape[2])
            ], dim=2)

        return (torch.stack([