From 4686c7cb78ab9e4a4f8880a5f23505bd58edc8f6 Mon Sep 17 00:00:00 2001 From: Qing Date: Fri, 18 Nov 2022 23:06:41 +0800 Subject: [PATCH] fix manga mask --- lama_cleaner/model/manga.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lama_cleaner/model/manga.py b/lama_cleaner/model/manga.py index 2fd73a8..577883a 100644 --- a/lama_cleaner/model/manga.py +++ b/lama_cleaner/model/manga.py @@ -104,7 +104,7 @@ class Manga(InpaintModel): mask = torch.from_numpy(mask[np.newaxis, :, :, :]).to(self.device) mask = mask.permute(0, 3, 1, 2) - mask = torch.where(mask > 0.5, torch.tensor(1.0), torch.tensor(0.0)) + mask = torch.where(mask > 0.5, 1.0, 0.0) noise = torch.randn_like(mask) gray_img = gray_img / 255 * 2 - 1.0