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

Merge cb820f02492a8ecb866bb41eecacafc5c9ca7611 into 50e7dd34d3b6e6bbab1d41e8068e1ddd19bd4d1b

Browse Source
This commit is contained in:
sovahc 2025-12-03 10:11:17 -07:00 committed by GitHub
commit 78db78ab28
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
nodes/image_nodes.py
View File

@ -2156,8 +2156,8 @@ class ReplaceImagesInBatch:
FUNCTION = "replace"
CATEGORY = "KJNodes/image"
DESCRIPTION = """
Replaces the images in a batch, starting from the specified start index,
with the replacement images.
Replaces the images in a batch, starting from the specified start index with step stride,
using the replacement images.
"""
@classmethod
@ -2165,6 +2165,7 @@ with the replacement images.
return {
"required": {
"start_index": ("INT", {"default": 1,"min": 0, "max": 4096, "step": 1}),
"step": ("INT", {"default": 1,"min": 1, "max": 4096, "step": 1}),
},
"optional": {
"original_images": ("IMAGE",),
@ -2174,14 +2175,14 @@ with the replacement images.
}
}
def replace(self, original_images=None, replacement_images=None, start_index=1, original_masks=None, replacement_masks=None):
def replace(self, original_images=None, replacement_images=None, start_index=1, step=1, original_masks=None, replacement_masks=None):
images = None
masks = None
if original_images is not None and replacement_images is not None:
if start_index >= len(original_images):
raise ValueError("ReplaceImagesInBatch: Start index is out of range")
end_index = start_index + len(replacement_images)
end_index = start_index + len(replacement_images) * step
if end_index > len(original_images):
raise ValueError("ReplaceImagesInBatch: End index is out of range")
@ -2189,7 +2190,7 @@ with the replacement images.
if original_images_copy.shape[2] != replacement_images.shape[2] or original_images_copy.shape[3] != replacement_images.shape[3]:
replacement_images = common_upscale(replacement_images.movedim(-1, 1), original_images_copy.shape[1], original_images_copy.shape[2], "lanczos", "center").movedim(1, -1)
original_images_copy[start_index:end_index] = replacement_images
original_images_copy[start_index:end_index:step] = replacement_images
images = original_images_copy
else:
images = torch.zeros((1, 64, 64, 3))
@ -2197,7 +2198,7 @@ with the replacement images.
if original_masks is not None and replacement_masks is not None:
if start_index >= len(original_masks):
raise ValueError("ReplaceImagesInBatch: Start index is out of range")
end_index = start_index + len(replacement_masks)
end_index = start_index + len(replacement_masks) * step
if end_index > len(original_masks):
raise ValueError("ReplaceImagesInBatch: End index is out of range")
@ -2205,7 +2206,7 @@ with the replacement images.
if original_masks_copy.shape[1] != replacement_masks.shape[1] or original_masks_copy.shape[2] != replacement_masks.shape[2]:
replacement_masks = common_upscale(replacement_masks.unsqueeze(1), original_masks_copy.shape[1], original_masks_copy.shape[2], "nearest-exact", "center").squeeze(0)
original_masks_copy[start_index:end_index] = replacement_masks
original_masks_copy[start_index:end_index:step] = replacement_masks
masks = original_masks_copy
else:
masks = torch.zeros((1, 64, 64))