nodes/lquesada-ComfyUI-Inpaint-CropAndStitch
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix fill_holes breaking mask at the borders

Browse Source
This commit is contained in:
Luis Quesada
2024-05-13 09:10:41 +02:00
parent 0f62cd0126
commit c4957772fb
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
inpaint_cropandstitch.py
+3 -4
View File
@@ -1,5 +1,4 @@
import comfy.utils
import math
import nodes
import numpy as np
import torch
@@ -93,7 +92,7 @@ class InpaintCrop:
mask_np = m.numpy()
binary_mask = mask_np > 0
struct = np.ones((5, 5))
closed_mask = binary_closing(binary_mask, structure=struct)
closed_mask = binary_closing(binary_mask, structure=struct, border_value=1)
filled_mask = binary_fill_holes(closed_mask)
output = filled_mask.astype(np.float32) * 255
output = torch.from_numpy(output)
@@ -115,8 +114,8 @@ class InpaintCrop:
if internal_upscale_factor < 0.999 or internal_upscale_factor > 1.001:
samples = image
samples = samples.movedim(-1, 1)
width = math.ceil(samples.shape[3] * internal_upscale_factor)
height = math.ceil(samples.shape[2] * internal_upscale_factor)
width = round(samples.shape[3] * internal_upscale_factor)
height = round(samples.shape[2] * internal_upscale_factor)
samples = comfy.utils.bislerp(samples, width, height)
effective_upscale_factor_x = float(width)/float(original_width)
effective_upscale_factor_y = float(height)/float(original_height)