From 0ecdb0af743d48461b23ba0e308dfe3466c6536a Mon Sep 17 00:00:00 2001 From: Qing Date: Tue, 27 Feb 2024 22:02:19 +0800 Subject: [PATCH] remove extra mask_blur --- iopaint/model/base.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/iopaint/model/base.py b/iopaint/model/base.py index f8efd8e..83354f2 100644 --- a/iopaint/model/base.py +++ b/iopaint/model/base.py @@ -404,7 +404,7 @@ class DiffusionInpaintModel(InpaintModel): def forward_pre_process(self, image, mask, config): if config.sd_mask_blur != 0: 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 @@ -412,7 +412,7 @@ class DiffusionInpaintModel(InpaintModel): if config.sd_match_histograms: result = self._match_histograms(result, image[:, :, ::-1], mask) - if config.sd_mask_blur != 0: - k = 2 * config.sd_mask_blur + 1 - mask = cv2.GaussianBlur(mask, (k, k), 0) + # if config.sd_mask_blur != 0: + # k = 2 * config.sd_mask_blur + 1 + # mask = cv2.GaussianBlur(mask, (k, k), 0) return result, image, mask