minor changes

2025-02-21 12:42:16 +02:00
parent 824996bfbb
commit d8263fc564
1 changed files with 2 additions and 2 deletions
@@ -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)