298977887/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2025-01-11 10:25:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow GrowMask node to work with batches (for AnimateDiff) (#1623)

Browse Source 
* Allow mask batches

This allows LatentCompositeMasked -node to work with AnimateDiff. I tried to keep old functionality too, unsure if it's correct, but both single mask and batch of masks seems to work with this change.

* Update nodes_mask.py
This commit is contained in:
Jukka Seppänen 2023-09-29 05:01:19 +03:00 committed by GitHub
parent 66756de100
commit 1c8ae9dbb2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
comfy_extras/nodes_mask.py
View File

@ -331,15 +331,14 @@ class GrowMask:
out = [] out = []
for m in mask: for m in mask:
output = m.numpy() output = m.numpy()
while expand < 0: for _ in range(abs(expand)):
if expand < 0:
output = scipy.ndimage.grey_erosion(output, footprint=kernel) output = scipy.ndimage.grey_erosion(output, footprint=kernel)
expand += 1 else:
while expand > 0:
output = scipy.ndimage.grey_dilation(output, footprint=kernel) output = scipy.ndimage.grey_dilation(output, footprint=kernel)
expand -= 1
output = torch.from_numpy(output) output = torch.from_numpy(output)
out.append(output) out.append(output)
return (torch.cat(out, dim=0),) return (torch.stack(out, dim=0),)