diff --git a/datasets/raise/extract_noise.py b/datasets/raise/extract_noise.py index 287ff4c..5f216a3 100755 --- a/datasets/raise/extract_noise.py +++ b/datasets/raise/extract_noise.py @@ -9,6 +9,7 @@ from tqdm import tqdm import csv import rawpy from utils import Color +import matplotlib.pyplot as plt imagesFolderPath = 'rafael/arw' imagesFolderPathFileName = imagesFolderPath.replace('/', '_') @@ -153,6 +154,9 @@ if denoiser == 'mean': colorMean = ((colorMean * numberOfImagesInColorMean) + imageNpArray) / (numberOfImagesInColorMean + 1) numberOfImagesInColorMean += 1 means[color] = colorMean + fileName = f'mean_{imagesFolderPathFileName}_{color}' + # Then use `merge_single_color_channel_images_according_to_bayer_filter.py` to consider all color channels, instead of saving this single color channel as an image. + saveNpArray(fileName, colorMean) for color in colors: mean = None