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

ImageResizeKJv2: output padded area as mask if no input mask to resize

Browse Source
This commit is contained in:
kijai 2025-07-15 14:57:57 +03:00
parent d9425173e7
commit fca4aa2a15
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
nodes/image_nodes.py
View File

@ -2510,6 +2510,8 @@ highest dimension.
if mask is not None: if mask is not None:
out_mask = mask.clone().to(device) out_mask = mask.clone().to(device)
else:
out_mask = None
if keep_proportion == "crop": if keep_proportion == "crop":
old_width = W old_width = W
@ -2573,6 +2575,11 @@ highest dimension.
out_mask = out_mask.unsqueeze(1).repeat(1, 3, 1, 1).movedim(1,-1) out_mask = out_mask.unsqueeze(1).repeat(1, 3, 1, 1).movedim(1,-1)
out_mask, _ = ImagePadKJ.pad(self, out_mask, pad_left, pad_right, pad_top, pad_bottom, 0, pad_color, "edge" if keep_proportion == "pad_edge" else "color") out_mask, _ = ImagePadKJ.pad(self, out_mask, pad_left, pad_right, pad_top, pad_bottom, 0, pad_color, "edge" if keep_proportion == "pad_edge" else "color")
out_mask = out_mask[:, :, :, 0] out_mask = out_mask[:, :, :, 0]
else:
B, H_pad, W_pad, _ = out_image.shape
out_mask = torch.ones((B, H_pad, W_pad), dtype=out_image.dtype, device=out_image.device)
out_mask[:, pad_top:pad_top+height, pad_left:pad_left+width] = 0.0
if unique_id and PromptServer is not None: if unique_id and PromptServer is not None:
try: try:
@ -2587,7 +2594,7 @@ highest dimension.
except: except:
pass pass
return(out_image.cpu(), out_image.shape[2], out_image.shape[1], out_mask.cpu() if mask is not None else torch.zeros(64,64, device=torch.device("cpu"), dtype=torch.float32)) return(out_image.cpu(), out_image.shape[2], out_image.shape[1], out_mask.cpu() if out_mask is not None else torch.zeros(64,64, device=torch.device("cpu"), dtype=torch.float32))
import pathlib import pathlib
class LoadAndResizeImage: class LoadAndResizeImage: