xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2025-12-09 12:54:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add back the commentary I deleted by mistake

Browse Source
This commit is contained in:
yvann-ba 2024-03-04 23:49:09 +01:00
parent 442c6a9127
commit 6ef1aa4a34
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes.py
View File

@ -772,17 +772,23 @@ class GrowMaskWithBlur:
current_expand += abs(incremental_expandrate) current_expand += abs(incremental_expandrate)
output = torch.from_numpy(output) output = torch.from_numpy(output)
if alpha < 1.0 and previous_output is not None: if alpha < 1.0 and previous_output is not None:
# Interpolate between the previous and current frame
output = alpha * output + (1 - alpha) * previous_output output = alpha * output + (1 - alpha) * previous_output
if decay < 1.0 and previous_output is not None: if decay < 1.0 and previous_output is not None:
# Add the decayed previous output to the current frame
output += decay * previous_output output += decay * previous_output
output = output / output.max() output = output / output.max()
previous_output = output previous_output = output
out.append(output) out.append(output)
if blur_radius != 0: if blur_radius != 0:
# Convert the tensor list to PIL images, apply blur, and convert back
for idx, tensor in enumerate(out): for idx, tensor in enumerate(out):
# Convert tensor to PIL image
pil_image = tensor2pil(tensor.cpu().detach())[0] pil_image = tensor2pil(tensor.cpu().detach())[0]
# Apply Gaussian blur
pil_image = pil_image.filter(ImageFilter.GaussianBlur(blur_radius)) pil_image = pil_image.filter(ImageFilter.GaussianBlur(blur_radius))
# Convert back to tensor
out[idx] = pil2tensor(pil_image) out[idx] = pil2tensor(pil_image)
blurred = torch.cat(out, dim=0) blurred = torch.cat(out, dim=0)
return (blurred, 1.0 - blurred) return (blurred, 1.0 - blurred)