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

add crop to ResizeImage

Browse Source
This commit is contained in:
kijai 2024-08-14 01:48:19 +03:00
parent eaed0d3593
commit 7e989daae3
1 changed files with 57 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
nodes/image_nodes.py
View File

@ -2,6 +2,7 @@ import numpy as np
import time import time
import torch import torch
import torch.nn.functional as F import torch.nn.functional as F
import torchvision.transforms as T
import random import random
import math import math
import os import os
@ -1506,6 +1507,7 @@ class ImageResizeKJ:
"width_input": ("INT", { "forceInput": True}), "width_input": ("INT", { "forceInput": True}),
"height_input": ("INT", { "forceInput": True}), "height_input": ("INT", { "forceInput": True}),
"get_image_size": ("IMAGE",), "get_image_size": ("IMAGE",),
"crop": (["disabled", "center", "top", "bottom", "left", "right"],),
} }
} }
@ -1525,8 +1527,10 @@ Keep proportions keeps the aspect ratio of the image, by
highest dimension. highest dimension.
""" """
def resize(self, image, width, height, keep_proportion, upscale_method, divisible_by, width_input=None, height_input=None, get_image_size=None): def resize(self, image, width, height, keep_proportion, upscale_method, divisible_by,
width_input=None, height_input=None, get_image_size=None, crop="disabled"):
B, H, W, C = image.shape B, H, W, C = image.shape
if width_input: if width_input:
width = width_input width = width_input
if height_input: if height_input:
@ -1553,15 +1557,46 @@ highest dimension.
if height == 0: if height == 0:
height = H height = H
if crop != "disabled":
if crop == "pad":
if H != W:
if H > W:
pad = (H - W) // 2
pad = (pad, 0, pad, 0)
elif W > H:
pad = (W - H) // 2
pad = (0, pad, 0, pad)
output = T.functional.pad(output, pad, fill=0)
else:
#crop_size = min(height, width)
x = (W-width) // 2
y = (H-height) // 2
if "top" in crop:
y = 0
elif "bottom" in crop:
y = H-height
elif "left" in crop:
x = 0
elif "right" in crop:
x = W-width
x2 = x+width
y2 = y+height
image = image[:, y:y2, x:x2, :]
else:
if divisible_by > 1 and get_image_size is None: if divisible_by > 1 and get_image_size is None:
width = width - (width % divisible_by) width = width - (width % divisible_by)
height = height - (height % divisible_by) height = height - (height % divisible_by)
image = image.movedim(-1,1)
scaled = common_upscale(image, width, height, upscale_method, 'disabled')
scaled = scaled.movedim(1,-1)
return(scaled, scaled.shape[2], scaled.shape[1],) image = image.movedim(-1,1)
image = common_upscale(image, width, height, upscale_method, "disabled")
image = image.movedim(1,-1)
return(image, image.shape[2], image.shape[1],)
class LoadAndResizeImage: class LoadAndResizeImage:
_color_channels = ["alpha", "red", "green", "blue"] _color_channels = ["alpha", "red", "green", "blue"]