Correct missing denoiser usage

This commit is contained in:
Benjamin Loison 2024-05-10 00:13:47 +02:00
parent 8523c02517
commit 73fb61b16d
Signed by: Benjamin_Loison
SSH Key Fingerprint: SHA256:BtnEgYTlHdOg1u+RmYcDE0mnfz1rhv5dSbQ2gyxW8B8

View File

@ -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]]