From 606821d45ec9a66e7ea5872311f67d8bf4da102c Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Wed, 3 Apr 2024 22:18:18 +0200 Subject: [PATCH] Add `show_mean_noise.py` --- datasets/raise/show_mean_noise.py | 7 +++++++ datasets/raise/split_and_compare_prnus_of_subgroups.py | 5 +++-- 2 files changed, 10 insertions(+), 2 deletions(-) create mode 100644 datasets/raise/show_mean_noise.py diff --git a/datasets/raise/show_mean_noise.py b/datasets/raise/show_mean_noise.py new file mode 100644 index 0000000..b194f01 --- /dev/null +++ b/datasets/raise/show_mean_noise.py @@ -0,0 +1,7 @@ +import numpy as np +import matplotlib.pyplot as plt + +npArray = np.load('mean.npy') +npArrayMin = npArray.min() +print(f'{npArrayMin=}') +plt.imsave('np_array.png', npArray - npArrayMin) \ No newline at end of file diff --git a/datasets/raise/split_and_compare_prnus_of_subgroups.py b/datasets/raise/split_and_compare_prnus_of_subgroups.py index f656b89..c31a0b7 100644 --- a/datasets/raise/split_and_compare_prnus_of_subgroups.py +++ b/datasets/raise/split_and_compare_prnus_of_subgroups.py @@ -3,6 +3,7 @@ import numpy as np import matplotlib.pyplot as plt from tqdm import tqdm from skimage.restoration import denoise_tv_chambolle +from skimage import img_as_float import sys sys.path.insert(0, '../../algorithms/image_utils/') @@ -28,8 +29,8 @@ for subgroupIndex in range(NUMBER_OF_SUBGROUPS): imagesPrnuEstimateNpArray = [] for imageFileName in tqdm(imagesFileNames[numberOfImagesPerSubgroup * subgroupIndex : numberOfImagesPerSubgroup * (subgroupIndex + 1)]): imagePath = f'{IMAGES_FOLDER}/{imageFileName}' - imagePil = Image.open(imagePath).convert('F') - imageNpArray = np.array(imagePil) + imagePil = Image.open(imagePath) + imageNpArray = img_as_float(np.array(imagePil)) imagePrnuEstimateNpArray = imageNpArray - denoise_tv_chambolle(imageNpArray, weight=0.2, channel_axis=-1) imagesPrnuEstimateNpArray += [imagePrnuEstimateNpArray]