From d8263fc56449310997d1708ca3f4e7f71b391514 Mon Sep 17 00:00:00 2001 From: unknown Date: Fri, 21 Feb 2025 12:42:16 +0200 Subject: [PATCH] minor changes --- color_transfer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/color_transfer.py b/color_transfer.py index 29f3f3f..86fd310 100644 --- a/color_transfer.py +++ b/color_transfer.py @@ -143,9 +143,9 @@ class PaletteTransferNode: color_matcher = ColorMatcher(distance_method) image_processor = ImagePostProcessor(gaussian_blur) - for img in image: + for img_tensor in image: # Prepare image - img = 255. * img.cpu().numpy() + img = 255. * img_tensor.cpu().numpy() # Convert color space converted_img, converted_colors = converter.convert_to_target_space(img, target_colors, color_space)