wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-04 10:40:30 +08:00
Code Issues Actions 17 Packages Projects Releases Wiki Activity

Merge aadccd28ca into bbe2c13a70

Browse Source
This commit is contained in:
Nicolas Stepien 2026-01-30 03:42:53 -05:00 committed by GitHub
commit 4361f56e1a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
comfy_extras/nodes_dataset.py
View File

@ -644,9 +644,13 @@ class ResizeImagesByShorterEdgeNode(ImageProcessingNode):
if w < h:
new_w = shorter_edge
new_h = int(h * (shorter_edge / w))
else:
elif h < w:
new_h = shorter_edge
new_w = int(w * (shorter_edge / h))
else:
# Square image: set both dimensions directly to avoid precision loss from float operations
new_w = shorter_edge
new_h = shorter_edge
img = img.resize((new_w, new_h), Image.Resampling.LANCZOS)
return pil_to_tensor(img)
@ -674,9 +678,13 @@ class ResizeImagesByLongerEdgeNode(ImageProcessingNode):
if w > h:
new_w = longer_edge
new_h = int(h * (longer_edge / w))
else:
elif h > w:
new_h = longer_edge
new_w = int(w * (longer_edge / h))
else:
# Square image: set both dimensions directly to avoid precision loss from float operations
new_w = longer_edge
new_h = longer_edge
img = img.resize((new_w, new_h), Image.Resampling.LANCZOS)
resized_images.append(pil_to_tensor(img))
return torch.cat(resized_images, dim=0)

11
comfy_extras/nodes_images.py
View File

@ -395,10 +395,15 @@ class ResizeAndPadImage(IO.ComfyNode):
scale_w = target_width / orig_width
scale_h = target_height / orig_height
scale = min(scale_w, scale_h)
new_width = int(orig_width * scale)
new_height = int(orig_height * scale)
# When aspect ratios match, scale directly to target to avoid precision loss from float multiplication
if scale_w == scale_h:
new_width = target_width
new_height = target_height
else:
scale = min(scale_w, scale_h)
new_width = int(orig_width * scale)
new_height = int(orig_height * scale)
image_permuted = image.permute(0, 3, 1, 2)