kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

fix manga mask

Browse Source
This commit is contained in:
Qing 2022-11-18 23:06:41 +08:00
parent 99af1dfe1c
commit 4686c7cb78
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lama_cleaner/model/manga.py
View File

@ -104,7 +104,7 @@ class Manga(InpaintModel):
mask = torch.from_numpy(mask[np.newaxis, :, :, :]).to(self.device) mask = torch.from_numpy(mask[np.newaxis, :, :, :]).to(self.device)
mask = mask.permute(0, 3, 1, 2) 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) noise = torch.randn_like(mask)
gray_img = gray_img / 255 * 2 - 1.0 gray_img = gray_img / 255 * 2 - 1.0