mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-02-10 21:42:37 +08:00
Update nodes_mask.py
This commit is contained in:
parent
738ff3f40f
commit
d793345ebb
@ -325,8 +325,8 @@ class GrowMask:
|
|||||||
def expand_mask(self, mask, expand, tapered_corners):
|
def expand_mask(self, mask, expand, tapered_corners):
|
||||||
c = 0 if tapered_corners else 1
|
c = 0 if tapered_corners else 1
|
||||||
kernel = np.array([[c, 1, c],
|
kernel = np.array([[c, 1, c],
|
||||||
[1, 1, 1],
|
[1, 1, 1],
|
||||||
[c, 1, c]])
|
[c, 1, c]])
|
||||||
mask = mask.reshape((-1, mask.shape[-2], mask.shape[-1]))
|
mask = mask.reshape((-1, mask.shape[-2], mask.shape[-1]))
|
||||||
out = []
|
out = []
|
||||||
for m in mask:
|
for m in mask:
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user