fix: resize img2img mask

타겟 이미지의 사이즈가 16의 배수가 아닐 때
This commit is contained in:
Dowon
2024-03-28 19:52:53 +09:00
parent 98043a8abd
commit 1ddace3d65

View File

@@ -660,6 +660,8 @@ class AfterDetailerScript(scripts.Script):
def inpaint_mask_filter(
img2img_mask: Image.Image, ad_mask: list[Image.Image]
) -> list[Image.Image]:
if ad_mask and img2img_mask.size != ad_mask[0].size:
img2img_mask = img2img_mask.resize(ad_mask[0].size, resample=images.LANCZOS)
return [mask for mask in ad_mask if has_intersection(img2img_mask, mask)]
@staticmethod