add the ImageTransitionTopToBottom node.

Author: zhangp365

README.md

@@ -125,6 +125,9 @@ This node is designed to generate a transition image between two images. The fir
 ## ImageTransitionLeftToRight
 This node is designed to generate a transition image between two images. The first image gradually slides to the right while the second image simultaneously appears from the left, creating a smooth transition effect.
 
+## ImageTransitionTopToBottom
+This node is designed to generate a transition image between two images. The first image gradually slides down while the second image simultaneously appears from the top, creating a smooth vertical transition effect.
+
 ## ImageMaskColorAverage
 This node is designed to calculate the average color of the image within the mask. It returns the decimal and hexadecimal values of the average color.
 

py/nodes_video.py

@@ -55,7 +55,9 @@ def adjust_frames(self, images: torch.Tensor, duration: float, fps: float, remov
             new_images = torch.cat([new_images, images[-1].unsqueeze(0).repeat(repeat_count, 1, 1, 1)], dim=0)
         return (new_images, len(new_images), fps)
 
-class ImageTransitionLeftToRight:
+class ImageTransitionBase:
+    """图像过渡效果的基类"""
+    
     def __init__(self):
         pass
 
@@ -66,6 +68,7 @@ def INPUT_TYPES(cls):
             "after_image": ("IMAGE",),
             "duration": ("FLOAT", {"default": 5.0, "min": 0.1, "max": 60.0, "step": 0.1}),
             "fps": ("FLOAT", {"default": 24.0, "min": 1.0, "max": 120.0, "step": 1.0}),
+            "bounce_back": ("BOOLEAN", {"default": False}),
             },
         }
 
@@ -74,52 +77,53 @@ def INPUT_TYPES(cls):
     FUNCTION = "create_transition"
     CATEGORY = "utils"
 
-    def create_transition(self, before_image: torch.Tensor, after_image: torch.Tensor, duration: float, fps: float):
-
+    def create_transition(self, before_image: torch.Tensor, after_image: torch.Tensor, duration: float, fps: float, bounce_back: bool = False):
         # 确保输入是单张图片，如果是批次则取第一张
         if len(before_image.shape) == 4 and before_image.shape[0] > 1:
             before_image = before_image[0:1]
         if len(after_image.shape) == 4 and after_image.shape[0] > 1:
             after_image = after_image[0:1]
 
-        # 获取目标尺寸（前图的尺寸）
-        _, target_width = before_image.shape[1:3]
-        
-        
-        adjusted_after = self.check_and_resizee_size(before_image, after_image)
+        # 调整后图尺寸以匹配前图
+        adjusted_after = self.check_and_resize_size(before_image, after_image)
 
         # 计算总帧数
         total_frames = int(duration * fps)
 
         # 创建过渡帧
         frames = []
 
-        for i in range(total_frames):
-            # 计算当前过渡位置 (0.0 到 1.0)
-            progress = i / (total_frames - 1) if total_frames > 1 else 1.0
-            
-            # 计算过渡线的x坐标
-            transition_x = int(target_width * progress)
-            
-            # 创建新帧
-            new_frame = torch.zeros_like(before_image)
+        if bounce_back:
+            # 如果启用回弹效果，先从左到右，再从右到左
+            # 前半段：从左到右
+            half_frames = total_frames // 2
+            for i in range(half_frames):
+                progress = i / (half_frames - 1) if half_frames > 1 else 1.0
+                new_frame = self.create_transition_frame(before_image, adjusted_after, progress)
+                frames.append(new_frame)
 
-            # 从左到右过渡：左侧显示后图，右侧显示前图
-            # 先填充整个前图
-            new_frame[0] = before_image[0]
-            
-            # 然后在左侧填充后图（覆盖前图）
-            if transition_x > 0:
-                new_frame[0, :, :transition_x,:] = adjusted_after[0, :,  :transition_x, :]
-            
-            frames.append(new_frame)
+            # 后半段：从右到左
+            for i in range(half_frames, total_frames):
+                progress = 1.0 - ((i - half_frames) / (total_frames - half_frames - 1) if total_frames - half_frames > 1 else 0.0)
+                new_frame = self.create_transition_frame(before_image, adjusted_after, progress)
+                frames.append(new_frame)
+        else:
+            # 正常的单向过渡
+            for i in range(total_frames):
+                progress = i / (total_frames - 1) if total_frames > 1 else 1.0
+                new_frame = self.create_transition_frame(before_image, adjusted_after, progress)
+                frames.append(new_frame)
 
         # 合并所有帧
         result = torch.cat(frames, dim=0)
 
         return (result, duration, fps)
 
-    def check_and_resizee_size(self, before_image, after_image):
+    def create_transition_frame(self, before_image: torch.Tensor, after_image: torch.Tensor, progress: float):
+        """创建过渡帧的抽象方法，子类需要实现"""
+        raise NotImplementedError("子类必须实现create_transition_frame方法")
+
+    def check_and_resize_size(self, before_image, after_image):
         # 获取目标尺寸（前图的尺寸）
         before_height, before_width = before_image.shape[1:3]
 
@@ -130,12 +134,10 @@ def check_and_resizee_size(self, before_image, after_image):
         if before_height == after_height and before_width == after_width:
             return after_image
 
-
         # 计算宽高比
         before_ratio = before_width / before_height
         after_ratio = after_width / after_height
 
-           
         logger.debug(f"before_image: {before_image.shape}, after_image: {after_image.shape}")
 
         # 调整后图尺寸，填充满目标尺寸（可能需要裁剪）
@@ -168,12 +170,60 @@ def check_and_resizee_size(self, before_image, after_image):
 
         return adjusted_after
 
+class ImageTransitionLeftToRight(ImageTransitionBase):
+    """从左到右的图像过渡效果"""
+    
+    def create_transition_frame(self, before_image: torch.Tensor, after_image: torch.Tensor, progress: float):
+        # 获取目标尺寸（前图的尺寸）
+        _, target_width = before_image.shape[1:3]
+        
+        # 计算过渡线的x坐标
+        transition_x = int(target_width * progress)
+        
+        # 创建新帧
+        new_frame = torch.zeros_like(before_image)
+        
+        # 从左到右过渡：左侧显示后图，右侧显示前图
+        # 先填充整个前图
+        new_frame[0] = before_image[0]
+        
+        # 然后在左侧填充后图（覆盖前图）
+        if transition_x > 0:
+            new_frame[0, :, :transition_x, :] = after_image[0, :, :transition_x, :]
+        
+        return new_frame
+
+class ImageTransitionTopToBottom(ImageTransitionBase):
+    """从上到下的图像过渡效果"""
+    
+    def create_transition_frame(self, before_image: torch.Tensor, after_image: torch.Tensor, progress: float):
+        # 获取目标尺寸（前图的尺寸）
+        target_height, _ = before_image.shape[1:3]
+        
+        # 计算过渡线的y坐标
+        transition_y = int(target_height * progress)
+        
+        # 创建新帧
+        new_frame = torch.zeros_like(before_image)
+        
+        # 从上到下过渡：上方显示后图，下方显示前图
+        # 先填充整个前图
+        new_frame[0] = before_image[0]
+        
+        # 然后在上方填充后图（覆盖前图）
+        if transition_y > 0:
+            new_frame[0, :transition_y, :, :] = after_image[0, :transition_y, :, :]
+        
+        return new_frame
+
 NODE_CLASS_MAPPINGS = {
     "FrameAdjuster": FrameAdjuster,
-    "ImageTransitionLeftToRight": ImageTransitionLeftToRight
+    "ImageTransitionLeftToRight": ImageTransitionLeftToRight,
+    "ImageTransitionTopToBottom": ImageTransitionTopToBottom
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
     "FrameAdjuster": "Frame Adjuster",
-    "ImageTransitionLeftToRight": "Image Transition Left to Right"
+    "ImageTransitionLeftToRight": "Image Transition Left to Right",
+    "ImageTransitionTopToBottom": "Image Transition Top to Bottom"
 }

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-utils-nodes"
 description = "Nodes:LoadImageWithSwitch, ImageBatchOneOrMore, ModifyTextGender, GenderControlOutput, ImageCompositeMaskedWithSwitch, ImageCompositeMaskedOneByOne, ColorCorrectOfUtils, SplitMask, MaskFastGrow, CheckpointLoaderSimpleWithSwitch, ImageResizeTo8x, MatchImageRatioToPreset, MaskFromFaceModel, MaskCoverFourCorners, DetectorForNSFW, DeepfaceAnalyzeFaceAttributes, VolcanoOutpainting, VolcanoImageEdit, etc."
-version = "1.3.2"
+version = "1.3.3"
 license = { file = "LICENSE" }
 dependencies = []