Skip to content

Commit

Permalink
Fix mask blur (#1907)
Browse files Browse the repository at this point in the history
* Fix mask blur

* Prioritize mask_blur_x over mask_blur

* Update mask blur sigma values
  • Loading branch information
catboxanon authored Aug 13, 2023
1 parent c542a8f commit 729926f
Showing 1 changed file with 16 additions and 2 deletions.
18 changes: 16 additions & 2 deletions scripts/controlnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,8 +163,22 @@ def prepare_mask(
mask = mask.convert("L")
if getattr(p, "inpainting_mask_invert", False):
mask = ImageOps.invert(mask)
if getattr(p, "mask_blur", 0) > 0:
mask = mask.filter(ImageFilter.GaussianBlur(p.mask_blur))

if hasattr(p, 'mask_blur_x'):
if getattr(p, "mask_blur_x", 0) > 0:
np_mask = np.array(mask)
kernel_size = 2 * int(2.5 * p.mask_blur_x + 0.5) + 1
np_mask = cv2.GaussianBlur(np_mask, (kernel_size, 1), p.mask_blur_x)
mask = Image.fromarray(np_mask)
if getattr(p, "mask_blur_y", 0) > 0:
np_mask = np.array(mask)
kernel_size = 2 * int(2.5 * p.mask_blur_y + 0.5) + 1
np_mask = cv2.GaussianBlur(np_mask, (1, kernel_size), p.mask_blur_y)
mask = Image.fromarray(np_mask)
else:
if getattr(p, "mask_blur", 0) > 0:
mask = mask.filter(ImageFilter.GaussianBlur(p.mask_blur))

return mask


Expand Down

0 comments on commit 729926f

Please sign in to comment.