xinyun/ComfyUI-KJNodes
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-KJNodes.git synced 2026-09-06 06:16:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2023-12-21 02:06:32 +02:00
parent 1448402b83
commit a8c1a707a9
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes.py
View File

@ -1147,13 +1147,15 @@ class ColorMatch:
def colormatch(self, image_ref, image_target, method): def colormatch(self, image_ref, image_target, method):
cm = ColorMatcher() cm = ColorMatcher()
image_ref = image_ref.cpu()
image_target = image_target.cpu()
batch_size = image_target.size(0) batch_size = image_target.size(0)
out = [] out = []
images_target = image_target.squeeze() images_target = image_target.squeeze()
images_ref = image_ref.squeeze() images_ref = image_ref.squeeze()
image_ref_np = images_ref.cpu().numpy() image_ref_np = images_ref.numpy()
images_target_np = images_target.cpu().numpy() images_target_np = images_target.numpy()
if image_ref.size(0) > 1 and image_ref.size(0) != batch_size: if image_ref.size(0) > 1 and image_ref.size(0) != batch_size:
raise ValueError("ColorMatch: Use either single reference image or a matching batch of reference images.") raise ValueError("ColorMatch: Use either single reference image or a matching batch of reference images.")