Upload bucket_proportional.py

bucket_proportional.py

@@ -0,0 +1,86 @@
+# Save this as ResizeToClosestProportional_alone.py in your ComfyUI custom_nodes directory
+
+import torch
+import logging
+from comfy.utils import common_upscale
+
+class ResizeToClosestProportional_alone:
+    upscale_methods = ["nearest-exact", "bilinear", "area", "bicubic", "lanczos"]
+    
+    @classmethod
+    def INPUT_TYPES(cls):
+        return {
+            "required": {
+                "images": ("IMAGE", ),
+                "base_resolution": ("INT", {
+                    "min": 64,
+                    "max": 1280,
+                    "step": 64,
+                    "default": 512,
+                    "tooltip": "Target resolution for the longest side, maintaining original proportions."
+                }),
+                "upscale_method": (cls.upscale_methods, {
+                    "default": "lanczos",
+                    "tooltip": "Upscale method to use"
+                }),
+                "crop": (["disabled", "center"],),
+            }
+        }
+
+    RETURN_TYPES = ("IMAGE", "INT", "INT")
+    RETURN_NAMES = ("images", "width", "height")
+    FUNCTION = "resize"
+    CATEGORY = "ImageProcessing"
+
+    def resize(self, images, base_resolution, upscale_method, crop):
+        """
+        Resize images maintaining original proportions to nearest size divisible by 16
+        
+        Args:
+            images: Input tensor of shape (B, H, W, C)
+            base_resolution: Target resolution for the longest side
+            upscale_method: Method to use for upscaling
+            crop: Whether to center crop or not
+            
+        Returns:
+            tuple: (resized_images, width, height)
+        """
+        # Get input dimensions
+        B, H, W, C = images.shape
+        aspect_ratio = H / W
+
+        # Determine which dimension is the longest and scale accordingly
+        if H > W:
+            # Height is the longest side
+            target_height = base_resolution
+            target_width = int(target_height / aspect_ratio)
+        else:
+            # Width is the longest side
+            target_width = base_resolution
+            target_height = int(target_width * aspect_ratio)
+
+        # Round to nearest multiple of 16
+        height = int(round(target_height / 16) * 16)
+        width = int(round(target_width / 16) * 16)
+
+        # Ensure minimum size of 64
+        height = max(64, height)
+        width = max(64, width)
+
+        logging.info(f"Resized to: {width}x{height} (maintaining original aspect ratio: {aspect_ratio:.2f})")
+
+        # Perform resize operation
+        resized_images = images.clone().movedim(-1, 1)  # Move channels to correct position
+        resized_images = common_upscale(resized_images, width, height, upscale_method, crop)
+        resized_images = resized_images.movedim(1, -1)  # Move channels back
+        
+        return (resized_images, width, height)
+
+# Node class mappings for ComfyUI
+NODE_CLASS_MAPPINGS = {
+    "ResizeToClosestProportional_alone": ResizeToClosestProportional_alone
+}
+
+NODE_DISPLAY_NAME_MAPPINGS = {
+    "ResizeToClosestProportional_alone": "Resize to Closest Proportional (Standalone)"
+}