chris.dumas/DumasNodes
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Optimize Dumas image compare conversion path

Browse Source
This commit is contained in:
Chris Dumas
2026-07-21 12:12:08 +00:00
parent a67ae5bf86
commit a7a7749068
1 changed files with 24 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

31
dumas_image_nodes.py
View File

@@ -7,6 +7,19 @@ from PIL import Image
import folder_paths
def _tensor_image_to_pil_image(tensor):
image_tensor = tensor[0]
if hasattr(image_tensor, "mul") and hasattr(image_tensor, "clamp"):
image_array = image_tensor.mul(255).clamp(0, 255)
if hasattr(image_array, "byte"):
image_array = image_array.byte()
image_array = image_array.cpu().numpy()
return Image.fromarray(image_array)
image_array = 255.0 * image_tensor.cpu().numpy()
return Image.fromarray(np.clip(image_array, 0, 255).astype(np.uint8))
class DumasImageCompareNode:
DESCRIPTION = (
"Dumas Image Compare shows the difference between two images directly on "
@@ -54,25 +67,29 @@ class DumasImageCompareNode:
}
def compare_images(self, image1=None, image2=None):
pairs = ((1, image1), (2, image2))
present = [(slot, tensor) for slot, tensor in pairs if tensor is not None]
present = []
if image1 is not None:
present.append((1, image1))
if image2 is not None:
present.append((2, image2))
results = []
if present:
first_tensor = present[0][1]
prefix = "dumas_compare" + self.prefix_append
first_image = first_tensor[0]
full_output_folder, filename, counter, subfolder, _ = folder_paths.get_save_image_path(
prefix,
self.output_dir,
first_tensor[0].shape[1],
first_tensor[0].shape[0],
first_image.shape[1],
first_image.shape[0],
)
join_path = os.path.join
for slot, tensor in present:
image_array = 255.0 * tensor[0].cpu().numpy()
image = Image.fromarray(np.clip(image_array, 0, 255).astype(np.uint8))
image = _tensor_image_to_pil_image(tensor)
file_name = f"{filename}_{counter:05}_.png"
image.save(
os.path.join(full_output_folder, file_name),
join_path(full_output_folder, file_name),
compress_level=self.compress_level,
)
results.append(