From 2f66e82f22426d5b66c12e2fa8d5678442c5221c Mon Sep 17 00:00:00 2001 From: Benjamin Loison Date: Fri, 29 Mar 2024 01:47:01 +0100 Subject: [PATCH] Show both RMS and RMS normalized --- algorithms/distance/rms_diff.py | 10 +++++++++- datasets/noise_free_test_images/estimate_prnu.py | 4 ++-- 2 files changed, 11 insertions(+), 3 deletions(-) diff --git a/algorithms/distance/rms_diff.py b/algorithms/distance/rms_diff.py index 7c0a944..ae512e6 100644 --- a/algorithms/distance/rms_diff.py +++ b/algorithms/distance/rms_diff.py @@ -14,5 +14,13 @@ def rmsDiffPil(im1, im2): map(lambda h, i: h*(i**2), h, range(256)) ) / (float(im1.size[0]) * im1.size[1])) -def rmsDiffNumpy(image0, image1): +def normalizeImage(image): + image = image - image.min() + image = image / image.max() + return image + +def rmsDiffNumpy(image0, image1, normalize = False): + if normalize: + image0 = normalizeImage(image0) + image1 = normalizeImage(image1) return np.sqrt(np.mean(np.square(image0 - image1))) \ No newline at end of file diff --git a/datasets/noise_free_test_images/estimate_prnu.py b/datasets/noise_free_test_images/estimate_prnu.py index b27e241..5ee7d91 100644 --- a/datasets/noise_free_test_images/estimate_prnu.py +++ b/datasets/noise_free_test_images/estimate_prnu.py @@ -58,6 +58,6 @@ for imageName in os.listdir(datasetPath): imagesPrnuEstimateNpArray += [imagePrnuEstimateNpArray] cameraPrnuEstimateNpArray = np.array(imagesPrnuEstimateNpArray).mean(axis = 0) -rms = rmsDiffNumpy(cameraPrnuEstimateNpArray, prnuNpArray) -showImageWithMatplotlib(cameraPrnuEstimateNpArray, f'Camera PRNU estimate\nRMS with actual one: {rms:.4f}') +#rms = rmsDiffNumpy(cameraPrnuEstimateNpArray, prnuNpArray, True) +showImageWithMatplotlib(cameraPrnuEstimateNpArray, f'Camera PRNU estimate\nRMS with actual one: {rmsDiffNumpy(cameraPrnuEstimateNpArray, prnuNpArray):.4f} (normalized RMS: {rmsDiffNumpy(cameraPrnuEstimateNpArray, prnuNpArray, True):.4f})')