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

ReplaceImagesInBatch: Allow replacing only masks without images

Browse Source
This commit is contained in:
kijai 2025-07-03 17:51:40 +03:00
parent ff49e1b01f
commit 1f653a8d96
1 changed files with 27 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
nodes/image_nodes.py
View File

@ -2007,25 +2007,43 @@ with the replacement images.
def INPUT_TYPES(s): def INPUT_TYPES(s):
return { return {
"required": { "required": {
"original_images": ("IMAGE",),
"replacement_images": ("IMAGE",),
"start_index": ("INT", {"default": 1,"min": 0, "max": 4096, "step": 1}), "start_index": ("INT", {"default": 1,"min": 0, "max": 4096, "step": 1}),
}, },
"optional": { "optional": {
"original_images": ("IMAGE",),
"replacement_images": ("IMAGE",),
"original_masks": ("MASK",), "original_masks": ("MASK",),
"replacement_masks": ("MASK",), "replacement_masks": ("MASK",),
} }
} }
def replace(self, original_images, replacement_images, start_index, original_masks=None, replacement_masks=None): def replace(self, original_images=None, replacement_images=None, start_index=1, original_masks=None, replacement_masks=None):
images = None images = None
if start_index >= len(original_images): masks = None
raise ValueError("GetImageRangeFromBatch: Start index is out of range")
end_index = start_index + len(replacement_images) if original_images is not None and replacement_images is not None:
if end_index > len(original_images): if start_index >= len(original_images):
raise ValueError("GetImageRangeFromBatch: End index is out of range") raise ValueError("ReplaceImagesInBatch: Start index is out of range")
end_index = start_index + len(replacement_images)
if end_index > len(original_images):
raise ValueError("ReplaceImagesInBatch: End index is out of range")
original_images_copy = original_images.clone()
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
images = original_images_copy
else:
images = torch.zeros((1, 64, 64, 3))
if original_masks is not None and replacement_masks is not None: 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)
if end_index > len(original_masks):
raise ValueError("ReplaceImagesInBatch: End index is out of range")
original_masks_copy = original_masks.clone() original_masks_copy = original_masks.clone()
if original_masks_copy.shape[1] != replacement_masks.shape[1] or original_masks_copy.shape[2] != replacement_masks.shape[2]: 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) replacement_masks = common_upscale(replacement_masks.unsqueeze(1), original_masks_copy.shape[1], original_masks_copy.shape[2], "nearest-exact", "center").squeeze(0)
@ -2033,15 +2051,8 @@ with the replacement images.
original_masks_copy[start_index:end_index] = replacement_masks original_masks_copy[start_index:end_index] = replacement_masks
masks = original_masks_copy masks = original_masks_copy
else: else:
masks = torch.zeros(1,64,64, device=original_images.device, dtype=original_images.dtype) masks = torch.zeros((1, 64, 64))
original_images_copy = original_images.clone()
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
images = original_images_copy
return (images, masks) return (images, masks)