From 73fb61b16d6e128e369bdfe85caca4e1372dd255 Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Fri, 10 May 2024 00:13:47 +0200 Subject: [PATCH] Correct missing `denoiser` usage --- ...ingle_color_channel_images_according_to_bayer_filter.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/datasets/raise/merge_with_same_color_scale_single_color_channel_images_according_to_bayer_filter.py b/datasets/raise/merge_with_same_color_scale_single_color_channel_images_according_to_bayer_filter.py index 7095b17..4de3581 100755 --- a/datasets/raise/merge_with_same_color_scale_single_color_channel_images_according_to_bayer_filter.py +++ b/datasets/raise/merge_with_same_color_scale_single_color_channel_images_according_to_bayer_filter.py @@ -6,8 +6,8 @@ import numpy as np PREFIX = 'means/mean_rafael_230424_' -def getImageByColor(color): - filePath = PREFIX + f'{color}.npy' +def getImageByColor(color, denoiser): + filePath = PREFIX + f'{denoiser}_{color}.npy' image = np.load(filePath) return image @@ -20,13 +20,12 @@ DENOISERS = [ ] for denoiser in DENOISERS: - singleColorChannelImages = {color: getImageByColor(color) for color in Color} + singleColorChannelImages = {color: getImageByColor(color, denoiser) for color in Color} multipleColorsImage = mergeSingleColorChannelImagesAccordingToBayerFilter(singleColorChannelImages) multipleColorsImageDenoisers[denoiser] = multipleColorsImage def getExtreme(extreme): vExtreme = extreme(list(multipleColorsImageDenoisers.values())) - #print(vExtreme) return vExtreme vMin, vMax = [getExtreme(extreme) for extreme in [np.min, np.max]]