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

remove extra mask_blur

Browse Source
This commit is contained in:
Qing 2024-02-27 22:02:19 +08:00
parent c7339bde1b
commit 0ecdb0af74
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
iopaint/model/base.py
View File

@ -404,7 +404,7 @@ class DiffusionInpaintModel(InpaintModel):
def forward_pre_process(self, image, mask, config): def forward_pre_process(self, image, mask, config):
if config.sd_mask_blur != 0: if config.sd_mask_blur != 0:
k = 2 * config.sd_mask_blur + 1 k = 2 * config.sd_mask_blur + 1
mask = cv2.GaussianBlur(mask, (k, k), 0)[:, :, np.newaxis] mask = cv2.GaussianBlur(mask, (k, k), 0)
return image, mask return image, mask
@ -412,7 +412,7 @@ class DiffusionInpaintModel(InpaintModel):
if config.sd_match_histograms: if config.sd_match_histograms:
result = self._match_histograms(result, image[:, :, ::-1], mask) result = self._match_histograms(result, image[:, :, ::-1], mask)
if config.sd_mask_blur != 0: # if config.sd_mask_blur != 0:
k = 2 * config.sd_mask_blur + 1 # k = 2 * config.sd_mask_blur + 1
mask = cv2.GaussianBlur(mask, (k, k), 0) # mask = cv2.GaussianBlur(mask, (k, k), 0)
return result, image, mask return result, image, mask