From 4ac75448a6a39c0dd2f2e0d1c25993597a2643fd Mon Sep 17 00:00:00 2001 From: Benjamin Loison <12752145+Benjamin-Loison@users.noreply.github.com> Date: Sat, 27 Apr 2024 17:54:16 +0200 Subject: [PATCH] Correct in theory the range of color values --- datasets/raise/split_and_compare_prnus_of_subgroups.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/datasets/raise/split_and_compare_prnus_of_subgroups.py b/datasets/raise/split_and_compare_prnus_of_subgroups.py index 65b7c70..91c5527 100755 --- a/datasets/raise/split_and_compare_prnus_of_subgroups.py +++ b/datasets/raise/split_and_compare_prnus_of_subgroups.py @@ -37,9 +37,15 @@ for subgroupIndex in tqdm(range(NUMBER_OF_SUBGROUPS), 'Subgroup'): subgroupIterativeMean.add(imagePrnuEstimateNpArray) subgroupPrnuEstimateNpArray += [subgroupIterativeMean.mean] - plt.imsave(f'prnu_subgroup_{subgroupIndex}.png', subgroupPrnuEstimateNpArray[-1]) subgroupsPrnuEstimatesNpArray += [subgroupPrnuEstimateNpArray] +mostImagesSubgroupPrnuEstimatesNpArray = [subgroupPrnuEstimatesNpArray[-1] for subgroupPrnuEstimatesNpArray in subgroupsPrnuEstimateNpArray] +minimum = np.min(mostImagesSubgroupPrnuEstimatesNpArray) +maximum = np.max(mostImagesSubgroupPrnuEstimatesNpArray) + +for subgroupIndex in range(NUMBER_OF_SUBGROUPS): + plt.imsave(f'prnu_subgroup_{subgroupIndex}.png', (subgroupsPrnuEstimateNpArray[subgroupIndex][-1] - minimum) / (maximum - minimum)) + rmss = [] for numberOfImagesIndex in range(len(numberOfImagesThresholds)): rms = rmsDiffNumpy(subgroupsPrnuEstimatesNpArray[0][numberOfImagesIndex], subgroupsPrnuEstimatesNpArray[1][numberOfImagesIndex])