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

Update image_nodes.py

Browse Source
This commit is contained in:
kijai 2024-10-15 13:44:19 +03:00
parent f123b36d28
commit 74c335b6ef
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes/image_nodes.py
View File

@ -1323,12 +1323,13 @@ class TransitionImagesMulti:
"transition_type": (["horizontal slide", "vertical slide", "box", "circle", "horizontal bar", "vertical bar", "horizontal door", "vertical door", "fade"],), "transition_type": (["horizontal slide", "vertical slide", "box", "circle", "horizontal bar", "vertical bar", "horizontal door", "vertical door", "fade"],),
"transitioning_frames": ("INT", {"default": 1,"min": 0, "max": 4096, "step": 1}), "transitioning_frames": ("INT", {"default": 1,"min": 0, "max": 4096, "step": 1}),
"blur_radius": ("FLOAT", {"default": 0.0,"min": 0.0, "max": 100.0, "step": 0.1}), "blur_radius": ("FLOAT", {"default": 0.0,"min": 0.0, "max": 100.0, "step": 0.1}),
"reverse": ("BOOLEAN", {"default": False}),
"device": (["CPU", "GPU"], {"default": "CPU"}), "device": (["CPU", "GPU"], {"default": "CPU"}),
}, },
} }
#transitions from matteo's essential nodes #transitions from matteo's essential nodes
def transition(self, inputcount, transitioning_frames, transition_type, interpolation, device, blur_radius, **kwargs): def transition(self, inputcount, transitioning_frames, transition_type, interpolation, device, blur_radius, reverse, **kwargs):
gpu = model_management.get_torch_device() gpu = model_management.get_torch_device()
@ -1339,6 +1340,8 @@ class TransitionImagesMulti:
mask = torch.zeros_like(images_1, device=images_1.device) mask = torch.zeros_like(images_1, device=images_1.device)
alpha = alpha.item() alpha = alpha.item()
if reverse:
alpha = 1 - alpha
if "horizontal slide" in transition_type: if "horizontal slide" in transition_type:
pos = round(width * alpha) pos = round(width * alpha)
@ -1409,7 +1412,6 @@ class TransitionImagesMulti:
return 1 - (1 - t) ** 4 return 1 - (1 - t) ** 4
def gaussian_blur(mask, blur_radius): def gaussian_blur(mask, blur_radius):
print(mask.device)
if blur_radius > 0: if blur_radius > 0:
kernel_size = int(blur_radius * 2) + 1 kernel_size = int(blur_radius * 2) + 1
if kernel_size % 2 == 0: if kernel_size % 2 == 0: