This commit is contained in:
Saquib Alam 2023-08-08 22:50:42 +05:30 committed by GitHub
parent 1d7bb7bae6
commit ea3063527b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1351,7 +1351,7 @@ class LoadImage:
mask1[0, :, :, 0] = mask
mask1[0, :, :, 1] = mask
mask1[0, :, :, 2] = mask
smooth_image = smooth_image * mask
smooth_image = smooth_image * mask1
mask = 1.0 - mask
else:
mask = torch.zeros((64,64), dtype=torch.float32, device="cpu")