Update nodes_mask.py

This commit is contained in:
kijai 2023-09-28 12:58:21 +03:00
parent 738ff3f40f
commit d793345ebb

View File

@ -325,8 +325,8 @@ class GrowMask:
def expand_mask(self, mask, expand, tapered_corners):
c = 0 if tapered_corners else 1
kernel = np.array([[c, 1, c],
[1, 1, 1],
[c, 1, c]])
[1, 1, 1],
[c, 1, c]])
mask = mask.reshape((-1, mask.shape[-2], mask.shape[-1]))
out = []
for m in mask: