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

View File

@ -1342,7 +1342,7 @@ class LoadImage:
image = np.array(image).astype(np.float32) / 255.0
image = torch.from_numpy(image)[None,]
smooth_image = image.copy()
smooth_image = image.detach().clone()
if 'A' in i.getbands():
mask = np.array(i.getchannel('A')).astype(np.float32) / 255.0
mask = torch.from_numpy(mask)