xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-05-11 04:53:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update image_nodes.py

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

45
nodes/image_nodes.py
View File

@ -1507,7 +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"],), "crop": (["disabled","center"],),
} }
} }
@ -1557,44 +1557,13 @@ highest dimension.
if height == 0: if height == 0:
height = H height = H
if crop != "disabled": if divisible_by > 1 and get_image_size is None:
if crop == "pad": width = width - (width % divisible_by)
if H != W: height = height - (height % divisible_by)
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 image = image.movedim(-1,1)
y2 = y+height image = common_upscale(image, width, height, upscale_method, crop)
image = image.movedim(1,-1)
image = image[:, y:y2, x:x2, :]
else:
if divisible_by > 1 and get_image_size is None:
width = width - (width % divisible_by)
height = height - (height % divisible_by)
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],) return(image, image.shape[2], image.shape[1],)