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