From 6b1200d99355e3700837c3365760dca0cb6f2db6 Mon Sep 17 00:00:00 2001 From: Jordan Thompson Date: Fri, 11 Aug 2023 00:38:25 -0700 Subject: [PATCH] Patch lucy padding --- WAS_Node_Suite.py | 18 +++++++----------- 1 file changed, 7 insertions(+), 11 deletions(-) diff --git a/WAS_Node_Suite.py b/WAS_Node_Suite.py index bf1faba..9a63478 100644 --- a/WAS_Node_Suite.py +++ b/WAS_Node_Suite.py @@ -2866,27 +2866,23 @@ class WAS_Lucy_Sharpen: kernel = np.ones((kernel_size, kernel_size), dtype=np.float32) / (kernel_size ** 2) sharpened_channels = [] + padded_image_array = np.pad(image_array, ((kernel_size, kernel_size), (kernel_size, kernel_size), (0, 0)), mode='edge') + for channel in range(3): - channel_array = image_array[:, :, channel] + channel_array = padded_image_array[:, :, channel] for _ in range(iterations): - padded_channel = np.pad(channel_array, kernel_size // 2, mode='edge') - - blurred_channel = convolve2d(padded_channel, kernel, mode='valid') + blurred_channel = convolve2d(channel_array, kernel, mode='same') ratio = channel_array / (blurred_channel + 1e-6) - - padded_ratio = np.pad(ratio, kernel_size // 2, mode='edge') - channel_array *= convolve2d(padded_ratio, kernel, mode='valid') + channel_array *= convolve2d(ratio, kernel, mode='same') sharpened_channels.append(channel_array) - sharpened_image_array = np.stack(sharpened_channels, axis=-1) - sharpened_image_array = np.clip(sharpened_image_array * 255.0, 0, 255).astype(np.uint8) + cropped_sharpened_image_array = np.stack(sharpened_channels, axis=-1)[kernel_size:-kernel_size, kernel_size:-kernel_size, :] + sharpened_image_array = np.clip(cropped_sharpened_image_array * 255.0, 0, 255).astype(np.uint8) sharpened_image = Image.fromarray(sharpened_image_array) return sharpened_image - - # IMAGE STYLE FILTER class WAS_Image_Style_Filter: